pytorch/xla

Pull Requests Commits

Disable ABI for the 2.0 release experimental build

JackCaoG committed 2 years ago

5e5445e5

Backport Python 3.9 Colab build support (#4790)

yeounoh committed 2 years ago

Verified 9d8446bf

Cherry-pick doc updates for PJRT to r2.0 (#4769)

will-cromar committed 2 years ago

Verified 09844348

Jack cao g/r20 backportdoc (#4752)

JackCaoG committed 2 years ago

Verified cf7995d9

Disable ABI for release (#4729)

JackCaoG committed 2 years ago

Verified 8c7394c0

backport dataloader thread from #4693 (#4727)

JackCaoG committed 2 years ago

Verified f642baf3

fix starting profiler server (#4689) (#4726)

JackCaoG committed 2 years ago

Verified 17f7da21

[PJRT] Cherry pick local rank changes to r2.0 release (#4712)

will-cromar committed 2 years ago

Verified 8af4ca19

Fix dynamo test no op (#4697) (#4701)

JackCaoG committed 2 years ago

Verified 42b7bfb9

Update env-setup.py (#4695) (#4700)

steventk-g committed 2 years ago

Verified 80b962d7

Patch in other regex updates (#4688)

steventk-g committed 2 years ago

Verified e2368274

Add regex matching for release branches (#4676) (#4677)

steventk-g committed 2 years ago

Verified 515b1065

Torch pin for release branch should be under root (#4675)

JackCaoG committed 2 years ago

Verified e1537681

Update 81058.diff

steventk-g committed 2 years ago

Verified a62fe903

Jack cao g/dynamo 2.0 cherry pick (#4653)

JackCaoG committed 2 years ago

Verified 32a320fa

Add pytorch patch file to revert https://github.com/pytorch/pytorch/pull/81058

steventk-g committed 2 years ago

Verified 768d7e1f

Delete torch_patches/torch_patches directory

steventk-g committed 2 years ago

Verified 44919d0e

Added pytorch patch file to revert PR #81058

steventk-g committed 2 years ago

Verified 86ddbe4c

[PJRT] Defer default device selection until we need to know the runtime (#4630) (#4640)

will-cromar committed 2 years ago

Verified a4a47369

Pin upstream 2.0 release (#4639)

steventk-g committed 2 years ago

Verified 5bef3f25

Disable failing `test_operations.py` tests on TPU (#4622)

will-cromar committed 2 years ago

66cc7370

Skip execution for extract_compiled_graph (#4612)

JackCaoG committed 2 years ago

973e8b49

Fix patch command in TPU CI (#4623)

will-cromar committed 2 years ago

ff8272b2

Update TF pin to 2/13 (#4615)

will-cromar committed 2 years ago

2c929681

Fix HLO dumping (#4619)

JackCaoG committed 2 years ago

e88dc21f

Remove reference to `torch._six` compat layer (#4613)

XuehaiPan committed 2 years ago

Verified d29eb67c

[FSDP] Use state_dict hook to save shard metadata for consolidating checkpoint (#4572)

YangFei1990 committed 2 years ago

Verified d13090c4

Support multihost SPMD execution (#4573)

jonb377 committed 2 years ago

Verified ced64569

[PJRT] Use TFRT TPU client by default (#4600)

will-cromar committed 2 years ago

Verified 16b4f03b

Add int as supported index tensor type (#4602)

ymwangg committed 2 years ago

Verified 249b8e69

Older