pytorch/xla

Pull Requests Commits

ManfeiBai committed 2 years ago

Verified 0e2c8cb3

Update dynamo number (#5660) (#5661)

JackCaoG committed 2 years ago

Verified 7942cfa7

lower NativeDropoutBackward (#5642) and lower native_dropout (#5643) (#5647)

ManfeiBai committed 2 years ago

Verified b8401372

Fix log spam when libtpu is loaded (#5619) (#5621)

ManfeiBai committed 2 years ago

Verified b62be6e5

Clean up README for 2.1 release branch. (#5618)

will-cromar committed 2 years ago

Verified 2d2c80cb

XLA pin downgrade to 20230825 (#5592) (#5604)

ManfeiBai committed 2 years ago

Verified 53fa74c7

Cherry-pick: Don't trigger CI build on release tag push (#5595)

will-cromar committed 2 years ago

Verified 5fcf79ce

Enable xla:gpu autocast for bfloat16 if not restricted (#5570) (#5591)

will-cromar committed 2 years ago

Verified 1e6c11d1

Improve bernoulli rng-bit-generation memory footprint (#5581) (#5589)

ManfeiBai committed 2 years ago

Verified fa5d132c

Fix release branch and tag patterns for GitHub Actions (#5587) (#5590)

will-cromar committed 2 years ago

Verified 79ce1c52

Move where clear pending IR is called to avoid crash (#5552) (#5582)

JackCaoG committed 2 years ago

Verified 3e2bf8aa

fixing num_local_processes typo (#5573) (#5579)

will-cromar committed 2 years ago

Verified cbb16022

[author: jluntamazon] Adding more explicit HLO lowering control by exposing LoweringContext… (#5431) (#5580)

wonjoo-wj committed 2 years ago

Verified 4ec0997d

Make cpu tensor on XLA dynamo backend a warning instead of error (#5549) (#5576)

JackCaoG committed 2 years ago

Verified c55ef0b6

Handle dynamo function without input (#5565) (#5577)

JackCaoG committed 2 years ago

Verified f0f9126b

Update torch pin (#5575)

wonjoo-wj committed 2 years ago

Verified 02858b90

Cherry pick `pjrt://` init method rename and doc updates (#5562)

will-cromar committed 2 years ago

Verified ee723327

[Pin Update] Version 20230826(#5527) (#5555)

ManfeiBai committed 2 years ago

Verified 2c07df94

Add resnet50-weight-quant colab notebook (#5407) (#5556)

wonjoo-wj committed 2 years ago

Verified 9bcf3f74

Update r2.1 wheel to be compatible with PyPI (#5550)

will-cromar committed 2 years ago

Verified 691838fb

Fix FSDP for Models with Frozen Weights (#5484) (#5539)

wonjoo-wj committed 2 years ago

Verified 89574d2d

Revert 2.1 terraform changes (#5537)

wonjoo-wj committed 2 years ago

Verified a1d3651f

Add option to unbundle libtpu (#5534) (#5536)

will-cromar committed 2 years ago

Verified 7a01093d

Enable megacore_dense by default (#5520) (#5531)

wonjoo-wj committed 2 years ago

Verified bd1958f3

Update pytorch git tag for r2.1 (#5529)

wonjoo-wj committed 2 years ago

Verified 3b34ae15

Disable cxx abi in ansible when building pt/xla for branch r2.0 (#5332)

vanbasten23 committed 2 years ago

3ac818ec

[SPMD] Propagate replicated output (#5508)

alanwaketan committed 2 years ago

Verified 5c4ce2a9

mark_sharding over a replicated tensor is allowed. (#5513)

yeounoh committed 2 years ago

Verified 6e90569d

Get original_traced_args as example_inputs. (#5511)

qihqi committed 2 years ago

Verified 33f1cd29

Update dynamo.md (#5509)

shauheen committed 2 years ago

Verified 808015d8

Older