pytorch/xla

Pull Requests Commits

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

Update API GUIDE to include multi host training and add some colors (#4706)

JackCaoG committed 2 years ago

Verified a6d186f7

Move dependencies on Tensorflow under third_party/xla_client and torch_xla/csrc to use TSL where possible (#4717)

ddunl committed 2 years ago

Verified c939de06

Enable PjRt multi-GPU tests in CI (#4718)

jonb377 committed 2 years ago

Verified 6d7cc081

[PJRT] Detect number of local TPU chips (#4720)

will-cromar committed 2 years ago

Verified 46d47d35

Revert "Support tf tf repo macros in external repos. (#4708)" (#4719)

alanwaketan committed 2 years ago

Verified 9e4b01af

Update assert to self.assertTrue (#4702)

JackCaoG committed 2 years ago

Verified fb092d27

Remove .torch_pin from the Functionalization PR (#4715)

alanwaketan committed 2 years ago

Verified 0466784d

Functionalization integration (#4158)

bdhirsh committed 2 years ago

Verified e49df835

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

stgpetrovic committed 2 years ago

Verified 5ca5686a

Lower UpsampleBilinear/Nearest2DBackward with scale factor on TPU (#4710)

lsy323 committed 2 years ago

Verified f594c29a

[PJRT] Remove start method from _run_singleprocess (#4709)

will-cromar committed 2 years ago

Verified d86b323d

Update GPU wheel link to also include CUDA 117 and 118 (#4705)

JackCaoG committed 2 years ago

Verified 5b5dd27f

Add test for dynamo and sgd optimizer (#4698)

JackCaoG committed 2 years ago

Verified ff15a8ac

remove windows related diff in patch (#4694)

lsy323 committed 2 years ago

Verified 89a9d7bc

Newer Older