pytorch/xla

Pull Requests Commits

[Functionalization] Properly skip GRU and LSTM tests

alanwaketan committed 2 years ago

4969ca92

Add detail about PJRT performance to `pjrt.md` (#4780)

will-cromar committed 2 years ago

Verified f7dea898

[Functionalization] Lower as_strided_scatter properly (#4770)

alanwaketan committed 2 years ago

Verified e2389eaf

Introduce a new flag to disable functionalization tensor wrapping (#4751)

wonjoo-wj committed 2 years ago

Verified fff028db

[FSDP] Add profiler to ImageNet (#4754)

alanwaketan committed 2 years ago

Verified d8125517

[PJRT] Update pjrt.md for new release (#4757)

will-cromar committed 2 years ago

Verified 902b52e0

Disable TestSyncFreeSGD.test_optimizer due to flakiness (#4766)

wonjoo-wj committed 2 years ago

Verified ab760a48

Remove a dynamic shape related comment that doesn't hold anymore. (#4433)

vanbasten23 committed 2 years ago

Verified cc41dad2

Make tpu ci run test for dynamic shape. (#4733)

vanbasten23 committed 2 years ago

Verified 2165ce93

Reenable CI coverage on master (#4761)

yeounoh committed 2 years ago

Verified 2e70c7bd

Return failing status code when TPU CI tests fail (#4750)

will-cromar committed 2 years ago

Verified c0e8e757

Re-enable dynamo tests with TODOs (#4746)

wonjoo-wj committed 2 years ago

Verified 8efc9675

Generate CI coverage metric in lcov format and upload (#4682)

yeounoh committed 2 years ago

Verified 63faa0dc

[PJRT] Update `torch.distributed` unit test to use PJRT (#4743)

will-cromar committed 2 years ago

Verified 5da15023

Fix typo in README.md (#4748)

ag8 committed 2 years ago

Verified b24267cf

Add public readme for torchdynamo (#4744)

JackCaoG committed 2 years ago

Verified 7ef2e1b5

POC of dynamicShape+functionalization integration (#4611)

vanbasten23 committed 2 years ago

Verified c6fca12d

Add feature to increase the number of host to device transfer threads (#4693)

chandrasekhard2 committed 2 years ago

Verified ea135c6d

Revert "Revert "Remove Bazel sha256 checksum from github archive link (#4731)" (#4737)" (#4738)

JackCaoG committed 2 years ago

Verified d44ad24b

Disable coverage on master (#4739)

yeounoh committed 2 years ago

Verified 613707c9

Revert "Remove Bazel sha256 checksum from github archive link (#4731)" (#4737)

JackCaoG committed 2 years ago

Verified 07b6fd9f

Remove Bazel sha256 checksum from github archive link (#4731)

huydhn committed 2 years ago

Verified d9d44867

Support tf tf repo macros in external repos. (#4730)

stgpetrovic committed 2 years ago

Verified ef45ce26

Update README (#4734)

JackCaoG committed 2 years ago

Verified 2a0b9df5

[FSDP] Add meta device and torchdistx deferred_init feature in FSDP (#4664)

Zach Zheng committed 2 years ago

Verified 2e0c00b9

[SPMD] Preserve parameter sharding with output data sharding (#4721)

yeounoh committed 2 years ago

Verified e2abcaf4

[PJRT] Replace XRT CI tests with PJRT (#4723)

will-cromar committed 2 years ago

Verified fd7f14ad

[FSDP] Fix pre-hook check when module is only used in forward (#4670)

zachzzc committed 2 years ago

Verified 50141f2e

fix linter (#4725)

JackCaoG committed 2 years ago

Verified 18674daf

Manually track type of SymNode (#4722)

ezyang committed 2 years ago

Verified add720fc

Older