pytorch/xla

Pull Requests Commits

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

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

Newer Older