pytorch/xla

Pull Requests Commits

Update xla_test_job.yaml

JackCaoG committed 2 years ago

Verified ec9bacc8

Move DS test to TFRT

JackCaoG committed 2 years ago

Verified 0c5014bb

Add dynamo test to TPU CI

JackCaoG committed 2 years ago

f868d446

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

Move development image trigger to the -releases GCP project and public docker repo. (#5041)

mateuszlewko committed 2 years ago

Verified bc28ca58

update gpu doc to use SPMD and new docker (#5036)

JackCaoG committed 2 years ago

Verified 93ff6aa4

Place XLAShard data on CPU on creation (#5032)

jonb377 committed 2 years ago

Verified 04640950

[SPMD] Output param sharding (#5018)

yeounoh committed 2 years ago

Verified b52b4a4c

[CI] Move lintcheck from Circle to GHA (#5035)

malfet committed 2 years ago

Verified 6bc1315b

Fix how we find C++ source files when formatting. (#5029)

stgpetrovic committed 2 years ago

Verified 87fb0ad4

Revert "Parallel tests (#4960) (#5030)" (#5033)

will-cromar committed 2 years ago

Verified 2c1b0efc

Add executeReplicated metrics (#5024)

JackCaoG committed 2 years ago

Verified 82319ca4

Parallel tests (#4960) (#5030)

stgpetrovic committed 2 years ago

Verified 1c57aeae

Update colab whl link for python 3.10 (#5020)

JackCaoG committed 2 years ago

Verified c6b7ca58

frmat cc files (#5028)

JackCaoG committed 2 years ago

Verified a1a77723

Enable SPMD + dynamo for inference (#5002)

JackCaoG committed 2 years ago

Verified dc057dcd

Update README.md with typo (#5026)

ManfeiBai committed 2 years ago

Verified 2d46002d

Older