pytorch/xla

Pull Requests Commits

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

CI test coverage (#4975)

stgpetrovic committed 2 years ago

Verified 38c80026

Replace COMMIT_ID with COMMIT_SHA (#4978)

mateuszlewko committed 2 years ago

Verified 82339d55

Add trigger for dev image with cuda support (#4976)

mateuszlewko committed 2 years ago

Verified 25fa0d0d

Add Python 3.10 nightly build (#4970)

will-cromar committed 2 years ago

Verified 6c3abb24

[PJRT] Cache world size and ordinal only when addressable_device_count == 1 (#4965)

will-cromar committed 2 years ago

Verified 5aa1ac3d

Revert "Parallel tests (#4960)" (#4968)

JackCaoG committed 2 years ago

Verified fd4a455f

Newer Older