pytorch/xla

Pull Requests Commits

Restrict the mark_sharding api to devicedata

JackCaoG committed 2 years ago

1f88b80f

[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

Update docker link (#5077)

JackCaoG committed 2 years ago

Verified 96bcfc9c

Move CI to GitHub Actions (#5045)

malfet committed 2 years ago

Verified 2b0abbaf

Tf pin update (#5023)

cowanmeg committed 2 years ago

Verified 3150573f

Remove google-cloud-cli and google apt repo (#5054)

mateuszlewko committed 2 years ago

Verified 7a1e9c74

[CI] Do not collect coverage on regular test jobs (#5049)

malfet committed 2 years ago

Verified 169bbea2

Change dev images trigger location to us-centeral1 (#5053)

mateuszlewko committed 2 years ago

Verified bc65243d

Add `yaml` to installation dependency in setup.py (#5044)

wonjoo-wj committed 2 years ago

Verified 9d9e089b

Older