pytorch/xla

Pull Requests Commits

[FSDP] Adopts XLA_DISABLE_FUNCTIONALIZATION (#4806)

alanwaketan committed 2 years ago

Verified cb02d52b

Update GITHUB_TORCH_XLA_BOT_TOKEN for doc_push (#4800)

yeounoh committed 2 years ago

Verified 749a0a35

[Functionalization] Harden XLA_DISABLE_FUNCTIONALIZATION (#4805)

alanwaketan committed 2 years ago

Verified aa532ff1

Add Terraform config with CloudBuild triggers for building docker images and wheels (#4604)

mateuszlewko committed 2 years ago

Verified f076b3fe

Update Colab docs to 2.0 (#4795)

steventk-g committed 2 years ago

Verified 4042c28b

Add XRT wheels (#4798)

steventk-g committed 2 years ago

Verified b3c4fbf1

Add colab release wheel (#4797)

steventk-g committed 2 years ago

Verified 02024e5d

Add support for XLA_DISABLE_FUNCTIONALIZATION flag (#4792)

wonjoo-wj committed 2 years ago

Verified b44325fa

[Functionalization] Properly lower slice_scatter (#4791)

alanwaketan committed 2 years ago

Verified 741fb36a

[Functionalization] Properly skip GRU and LSTM tests (#4789)

alanwaketan committed 2 years ago

Verified a6318d80

Update readme with r2.0 (#4783)

steventk-g committed 2 years ago

Verified f635b369

[Functionalization] detach_copy use set_ instead of _copy_from (#4785)

alanwaketan committed 2 years ago

Verified 05e947a3

Enable base_image=ubuntu:focal (#4787)

yeounoh committed 2 years ago

Verified a8aa4a32

[Functionalization] Lower diagonal_scatter properly (#4784)

alanwaketan committed 2 years ago

Verified fcef5273

Fix flaky test (#4782)

ymwangg committed 2 years ago

Verified f51e4ef5

Handle sparse PjRtDevice IDs (#4219)

jonb377 committed 2 years ago

Verified faca0a1f

* Enable whl build with py3.9 (#4778)

yeounoh committed 2 years ago

Verified bf0e6ee5

Add detail about PJRT performance to `pjrt.md` (#4780)

will-cromar committed 2 years ago

Verified f7dea898

[Functionalization] Lower as_strided_scatter properly (#4770)

alanwaketan committed 2 years ago

Verified e2389eaf

Introduce a new flag to disable functionalization tensor wrapping (#4751)

wonjoo-wj committed 2 years ago

Verified fff028db

[FSDP] Add profiler to ImageNet (#4754)

alanwaketan committed 2 years ago

Verified d8125517

[PJRT] Update pjrt.md for new release (#4757)

will-cromar committed 2 years ago

Verified 902b52e0

Disable TestSyncFreeSGD.test_optimizer due to flakiness (#4766)

wonjoo-wj committed 2 years ago

Verified ab760a48

Remove a dynamic shape related comment that doesn't hold anymore. (#4433)

vanbasten23 committed 2 years ago

Verified cc41dad2

Make tpu ci run test for dynamic shape. (#4733)

vanbasten23 committed 2 years ago

Verified 2165ce93

Reenable CI coverage on master (#4761)

yeounoh committed 2 years ago

Verified 2e70c7bd

Return failing status code when TPU CI tests fail (#4750)

will-cromar committed 2 years ago

Verified c0e8e757

Re-enable dynamo tests with TODOs (#4746)

wonjoo-wj committed 2 years ago

Verified 8efc9675

Generate CI coverage metric in lcov format and upload (#4682)

yeounoh committed 2 years ago

Verified 63faa0dc

[PJRT] Update `torch.distributed` unit test to use PJRT (#4743)

will-cromar committed 2 years ago

Verified 5da15023

Older