pytorch/xla

Pull Requests Commits

update gpu doc to use SPMD and new docker

JackCaoG committed 2 years ago

b1fd712f

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

Fix minor typos (#4972)

shahyash10 committed 2 years ago

Verified 6637b7f6

Add devcontainer config (#4966)

mateuszlewko committed 2 years ago

Verified 04c4cd63

Fix coverage for C++ in XLA. (#4984)

stgpetrovic committed 2 years ago

Verified 044425f6

Rearrange coverage config (#4982)

stgpetrovic committed 2 years ago

Verified 4440a5c2

Add instructions for creating non-master branch build trigger (#4969)

mateuszlewko committed 2 years ago

Verified 66c382ea

Older