pytorch/xla

Pull Requests Commits

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

[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

Newer Older