pytorch/xla

Pull Requests Commits

Propagates requires_grad over to AllReduce output

yeounoh committed 1 year ago

1fad0a89

Pin to pytorch#117182 (tags/v2.2.0-rc8) (#6335)

yeounoh committed 1 year ago

Verified dce624c3

Update 2.2 release README.md (#6311)

will-cromar committed 1 year ago

Verified 4a26144b

Correct order of constant vs args (#6304)

qihqi committed 1 year ago

Verified ce52ef33

Reenable ZeRO1 test for GPU to enable coverage for reduce-scatter/all-gather (#6278)

jeffhataws committed 1 year ago

Verified a9637349

[Cherry-pick]Promote int to float for acos operation(consistent with Pytorch) (#6224) (#6238)

ManfeiBai committed 1 year ago

Verified 7c46e4c5

[cherry-pick]Fix zero-1 bug for inferring local ranks (#5936) (#6237)

ManfeiBai committed 1 year ago

Verified 10d52690

Pin expecttest version (#6221) (#6228)

zpcore committed 1 year ago

Verified 09b13ef1

[Backport] Add default v5 flags (#6204)

jonb377 committed 1 year ago

Verified 53b13bbe

Introduce FSDPv2 (#6187)

alanwaketan committed 1 year ago

Verified 6860e307

Fix precision issue of pow(int, float) (#6103) (#6172)

qihqi committed 1 year ago

Verified a2a7ff64

Remove redundant log message (#6056) (#6185)

qihqi committed 1 year ago

Verified 94c07f7e

[Backport] Don't explcitly include virtual device in hash (#6167)

jonb377 committed 1 year ago

Verified dbfe5c35

[SPMD] Support mark_shard scalar tensors (#6158)

alanwaketan committed 1 year ago

Verified 98df3d1a

[Cherry-pick] Do manual type conversion for isnan op (#6110) (#6173)

wonjoo-wj committed 1 year ago

Verified 1cba38fe

Handled lifted constants in torch export. (#6111) (#6164)

qihqi committed 1 year ago

Verified 8b3d7a1f

[SPMD] Add apply_backward_optimization_barrier (#6157)

alanwaketan committed 1 year ago

Verified e854d07b

Make r2.2 release wheel compatible with PyPI (#6145)

will-cromar committed 2 years ago

Verified 419dd870

[Backport] Fix bad error message when `PjRtComputationClient` throws exception (#6144)

will-cromar committed 2 years ago

Verified dec978dc

Remove flaky test (#6120)

jonb377 committed 2 years ago

Verified 41d0fe58

[Backport]Handle negative dim for Diagonal Scatter (#6123) (#6129)

JackCaoG committed 2 years ago

Verified f87e9718

[Backport]Add out-of-place all-gather coalesced (#6059) (#6108)

JackCaoG committed 2 years ago

Verified 9a54abab

Update Troubleshotting doc with new PT_XLA_DEBUG (#6039) (#6095)

JackCaoG committed 2 years ago

Verified 53579b1d

[Backport]Add out-of-place reduce-scatter coalescing (#6109)

JackCaoG committed 2 years ago

Verified d1a7e9a0

Increase delay in trace_detached test (#6091)

jonb377 committed 2 years ago

Verified cd186bcd

[Backpoort]Move 0-dimensional tensors to CPU before copying to XLA. (#6071) (#6094)

JackCaoG committed 2 years ago

Verified c2cacdab

[Cherry-pick] [OpenXLA-pin] Update OpenXLA-pin to Nov29 (#6053) (#6079)

ManfeiBai committed 2 years ago

Verified 17cf6741

Add documentation and python API for persistent cache (#6074)

jonb377 committed 2 years ago

Verified a61ddef3

[Backport] Move CUDA tensors to CPU before moving to XLA (#6066)

will-cromar committed 2 years ago

Verified f77d1fac

[Backport]Add a spmd for application to pass in topology (#5931) (#6054)

JackCaoG committed 2 years ago

Verified 45742d2d

Older