pytorch/xla

Pull Requests Commits

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

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

Newer Older