pytorch/xla

Pull Requests Commits

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

In debug mode, add debug symbols to everything but tf (#5019)

stgpetrovic committed 2 years ago

Verified 30b2aba0

Disable cmake macros (#5021)

stgpetrovic committed 2 years ago

Verified e9a93333

Expose local shards as XLAShard (#5016)

jonb377 committed 2 years ago

Verified 56960f3f

Apply sharding when creating tensors data in data loader (#5015)

jonb377 committed 2 years ago

Verified f34f0b0a

Symintify fill_ (#4876)

vanbasten23 committed 2 years ago

Verified 032a4f3c

Update TPU devcontainer dependencies (#5013)

will-cromar committed 2 years ago

Verified 82354c35

Revert "Apply sharding when creating tensors data in data loader (#4995)" (#5012)

JackCaoG committed 2 years ago

Verified 6e9b663c

Apply sharding when creating tensors data in data loader (#4995)

jonb377 committed 2 years ago

Verified ce1054fc

Expose SharingUtil functions to generate global index slices (#5003)

jonb377 committed 2 years ago

Verified ede1ad0e

Add support for in place operation for dynamo bridge (#4992)

JackCaoG committed 2 years ago

Verified 2855373c

[Bazel] Update README.md for Bazel build wheels (#4996)

ManfeiBai committed 2 years ago

Verified 2b23b847

Wrap implicitly-replicated data in PjRtShardedData (#4971)

jonb377 committed 2 years ago

Verified 50144950

Clarify placement of release env variables (#4997)

mateuszlewko committed 2 years ago

Verified e0b635a2

Update tf pin to 202305 (#4973)

vanbasten23 committed 2 years ago

Verified 72d2874d

Set some helpful defaults in TPU devcontainer (#4991)

will-cromar committed 2 years ago

Verified 0268a5f6

[PJRT] Reduce the number of CPU transfers in rendezvous (#4990)

will-cromar committed 2 years ago

Verified 40e6c591

Newer Older