pytorch/xla

Pull Requests Commits

Revert "Revert "Adopt new changes in upstream""

alanwaketan committed 2 years ago

4c115053

Revert "Adopt new changes in upstream"

alanwaketan committed 2 years ago

1fe2a949

Adopt new changes in upstream

alanwaketan committed 2 years ago

4412fc22

Workaround all_gather graph breaks

alanwaketan committed 2 years ago

87df88c6

Reset token when mark_step

alanwaketan committed 2 years ago

680d7975

alanwaketan committed 2 years ago

e67bb04c

alanwaketan committed 2 years ago

4e342705

Initial support for all_reduce

alanwaketan committed 2 years ago

c55bfb80

Lower view_as_complex_copy and view_as_real_copy (#4887)

wonjoo-wj committed 2 years ago

Verified a47a7818

Use 'HEAD' instead of master for PyTroch clone (#4890)

mateuszlewko committed 2 years ago

Verified e5b7cb45

Update for SymInt API changes (#4882)

ezyang committed 2 years ago

Verified d7b0bdf6

Update pytorch branch for Terraform setup. (#4889)

mateuszlewko committed 2 years ago

Verified e73bfa8a

Move Terraform and Ansible setup into the `/infra` directory. (#4881)

mateuszlewko committed 2 years ago

Verified 28456ff3

Add documentation for Terraform setup (#4880)

mateuszlewko committed 2 years ago

Verified dbebb293

Revert experimental files deleted in github.com/pytorch/xla/pull/4842 (#4884)

mateuszlewko committed 2 years ago

Verified 35324bc9

Add missing file argument to clang-format-7 error message (#4883)

ezyang committed 2 years ago

Verified acf59f96

Configure Terrafrom to build all required artifacts in the new GCP project for releases (#4842)

mateuszlewko committed 2 years ago

Verified 47148b33

Update README.md (#4873)

JackCaoG committed 2 years ago

Verified b248e21f

Create a separate dynamic shape test folder (#4866)

vanbasten23 committed 2 years ago

Verified f763d0d2

Multinomial op lowering (#4848)

cowanmeg committed 2 years ago

Verified b437a331

Verify if abs support dynamism. (#4858)

vanbasten23 committed 2 years ago

Verified adb32a2e

[PJRT] Add example Kaggle notebooks for PyTorch/XLA 2.0 on TPU VM (#4860)

will-cromar committed 2 years ago

Verified 27fc8b28

Add flakiness to kokoro (#4855)

steventk-g committed 2 years ago

Verified 8cecc6f3

[PJRT] Improve `HasValue()` error message (#4857)

will-cromar committed 2 years ago

Verified 3d794c1d

Add C++ test coverage (#4812)

steventk-g committed 2 years ago

Verified 29a8bf25

add tests for clone. (#4846)

vanbasten23 committed 2 years ago

Verified 9268af8a

[PJRT] Count TPU chips by PCI IDs (#4845)

will-cromar committed 2 years ago

Verified afd614cd

Fix XLASymNodeImpl::has_hint (#4763)

vanbasten23 committed 2 years ago

Verified 1217d33b

Return early if comparing 2 identical symints (#4829)

vanbasten23 committed 2 years ago

Verified 52a0b671

Update README.md in .kokoro (#4833)

yeounoh committed 2 years ago

Verified 8e1d3ae9

Older