pytorch/xla

Pull Requests Commits

Remove duplicate shard exposure API (#5122)

jonb377 committed 2 years ago

Verified 83c48711

Delete unused dep `"@org_tensorflow//tensorflow/core/kernels:data_flow",` from TF (#5093)

ManfeiBai committed 2 years ago

Verified d21697da

Move PJRT Python APIs out of `torch_xla.experimental.*` (#5011)

will-cromar committed 2 years ago

Verified 7f4d1903

Add load_local_shards to XLAShardedTensor (#5047)

jonb377 committed 2 years ago

Verified 864e85df

Update CONTRIBUTING.md (#5119)

JackCaoG committed 2 years ago

Verified 46efb75c

Enable dynamism on xla::add (#4868)

vanbasten23 committed 2 years ago

Verified f2123949

[test] Delete `tensor_shape` and test (#5117)

ManfeiBai committed 2 years ago

Verified 07f21d65

Move TPUCI to TFRT for all tests. (#5098)

JackCaoG committed 2 years ago

Verified efccdad0

[SPMD] Support partial replication. (#4788)

yeounoh committed 2 years ago

Verified 85a4f141

fix a typo (#5113)

wbmc committed 2 years ago

Verified 04ade68e

[SPMD] index local shards correctly in the execution arguments for replication. (#5110)

yeounoh committed 2 years ago

Verified 2078d797

[CI] Pass CXX_ABI to cpp tests (#5107)

malfet committed 2 years ago

Verified 3f076072

Fix reference to ComputationClient::Get() (#5111)

will-cromar committed 2 years ago

Verified 09d6775d

Break dependency cycles within `:computation_client` (#5102)

will-cromar committed 2 years ago

Verified 843c850c

[SPMD] API for getting attributes for global devices (#5091)

khatwanimohit committed 2 years ago

Verified fcf8d438

Checkout Ansible setup at origin's master, not local, untracked master (#5106)

mateuszlewko committed 2 years ago

Verified c9457fd3

Revert "Disable cmake macros (#5021)" (#5105)

malfet committed 2 years ago

Verified 98494504

Add artifacts.auto.tfvars for 2.0 with python 3.10 (#5103)

wonjoo-wj committed 2 years ago

Verified 7a6aa398

Add isvalid check for return futures (#5100)

JackCaoG committed 2 years ago

Verified 3829437b

Parallel tests (#5066)

stgpetrovic committed 2 years ago

Verified 939f8961

Breaking depenency cycles. (#5065)

stgpetrovic committed 2 years ago

Verified d26e12be

[Dynamo] Refine CPU fallback for TD+XLA (#5000)

wonjoo-wj committed 2 years ago

Verified 3aeeb838

[OpenXLA]Delete conv_ops and conv_grad_op deps from tensorflow (#5090)

ManfeiBai committed 2 years ago

Verified 173e71ab

Use consistent precision types in sharded data replication. (#5083)

yeounoh committed 2 years ago

Verified 5c95ad2c

pass the PJRTFuture to lambda directly (#5086)

JackCaoG committed 2 years ago

Verified dd5731f9

Add more sharding unit tests (#5055)

wonjoo-wj committed 2 years ago

Verified a10c4504

Move the Lock acquireing to the end of ExecuteComputation and Execute… (#5078)

JackCaoG committed 2 years ago

Verified 08fb92cd

Fix minor issues with GHA workflow (#5081)

malfet committed 2 years ago

Verified 9c7d2ade

Checkout Ansible setup at current Cloud Build commit instead of master (#5080)

mateuszlewko committed 2 years ago

Verified 3fc38cb9

Add dev container config for external contributors (#5064)

mateuszlewko committed 2 years ago

Verified d601078d

Newer Older