pytorch/xla

Pull Requests Commits

Update README.md

JackCaoG committed 2 years ago

Verified 984aeb7f

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

Enable Kokoro integration testing (#4801)

yeounoh committed 2 years ago

Verified ce43fb1a

Catch info from metadata (#4724)

ManfeiBai committed 2 years ago

Verified 4d67a04e

Use ToXlaTensorArena for sharded data loader (#4822)

jonb377 committed 2 years ago

Verified 4dab8905

Add dev packages to the dev image. (#4826)

stgpetrovic committed 2 years ago

Verified 64e070cb

Return instead of exit when failing install_deps_pytorch_xla (#4764)

huydhn committed 2 years ago

Verified 5444e06e

Apply gradient checkpointing to spmd ResNet benchmarking script (#4816)

yeounoh committed 2 years ago

Verified abccaab4

[Profiler] Introduce trace_me decorator (#4809)

alanwaketan committed 2 years ago

Verified 3f3d01e7

Update python to 3.9 for GPU colab wheels (#4811)

steventk-g committed 2 years ago

Verified ae4bc3f9

[FSDP] Don't use detach in _rebuild_full_params (#4813)

alanwaketan committed 2 years ago

Verified 2b16a718

[SPMD] asynchronous data loading with ParallelLoader (#4808)

yeounoh committed 2 years ago

Verified 46e62380

Properly lower select_scatter (#4793)

wonjoo-wj committed 2 years ago

Verified a5d39687

[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

Older