pytorch/xla

Pull Requests Commits

alanwaketan committed 2 years ago

11ca2027

[SPMD] Properly skip tests on TPU V2

alanwaketan committed 2 years ago

8efb0b53

[SPMD] Fix HybridMesh ordering (#5478)

alanwaketan committed 2 years ago

Verified b8b48451

Don't move full tensor to device in deferred_init (#4819)

Zach Zheng committed 2 years ago

Verified c57370dc

Enable PJRT C API Client and other changes for Neuron (#5428)

aws-kingrj committed 2 years ago

Verified daed1d0a

Introduce `torch_xla.runtime.use_spmd()` (#5474)

yeounoh committed 2 years ago

Verified 462cb0f4

Enabling sharding device data IR (#5475)

JackCaoG committed 2 years ago

Verified 9a5fdf39

Fix the input sharding for dynamo (#5469)

JackCaoG committed 2 years ago

Verified bf997819

bump the timeout for CI (#5470)

JackCaoG committed 2 years ago

Verified 76495f0b

Fix data type in Pow with Scalar base and Tensor exponent (#5467)

lsy323 committed 2 years ago

Verified 909f28fa

update remote cache key in ansible (#5463)

lsy323 committed 2 years ago

Verified 028df4da

[GPU]Update README.md with wheel/docker for CUDA12.0 and deprecate CUDA11.7 (#5443)

ManfeiBai committed 2 years ago

Verified 95708f2b

Add missing includes (#5434)

peterbell10 committed 2 years ago

Verified c3c16cca

Neuron import hook (#5429)

aws-kingrj committed 2 years ago

Verified 6fa7770f

skip flaky test (#5459)

JackCaoG committed 2 years ago

Verified 5734ab8f

Update XLA pin to 2023/08/08 (#5427)

JackCaoG committed 2 years ago

Verified 0b311d00

Remove Docker image override from dev image (#5456)

will-cromar committed 2 years ago

Verified 941a4c48

Cache sharded placeholder for dynamo execution (#5446)

JackCaoG committed 2 years ago

Verified ee6ef4d6

Update to GCC 10 (#5451)

will-cromar committed 2 years ago

Verified ca2b4aac

Remove base image override from TPU CI build (#5453)

will-cromar committed 2 years ago

Verified 63febd9e

parallelize SPMD inputhandler and GetDataShards (#5447)

JackCaoG committed 2 years ago

Verified 481d0b04

Revert "Install gcc-10 (#5450)" (#5452)

will-cromar committed 2 years ago

Verified 493bb734

Install gcc-10 (#5450)

qihqi committed 2 years ago

Verified 65b7639c

Revert "Update gcc to 10. (#5445)" (#5449)

lsy323 committed 2 years ago

Verified 55249edd

Update the random seed for every dynamo execution (#5444)

JackCaoG committed 2 years ago

Verified 13f9382c

Update gcc to 10. (#5445)

qihqi committed 2 years ago

Verified 454e916e

Move the C++ test build to CI build job instead of test job (#5442)

JackCaoG committed 2 years ago

Verified 4ef4dd34

Bugfix + add more test for llama (#5439)

qihqi committed 2 years ago

Verified a2a4ff4c

[PJRT] Support `torchrun` with `pjrt://` `init_method` (#5438)

will-cromar committed 2 years ago

Verified 9db8fac4

Update fix_includes.sh (#5441)

aazzolini committed 2 years ago

Verified 5b88b5f8

Older