pytorch/xla

Pull Requests Commits

fix review comments

JackCaoG committed 2 years ago

0f4417d7

make XLA_DYNAMO_INPUT_SHARDING_CHECK_THREASHOLD configable

JackCaoG committed 2 years ago

ca235465

handle the cpu test

JackCaoG committed 2 years ago

cd29929f

add option to skip checking input sharding after x runs

JackCaoG committed 2 years ago

b9ec261b

only get sharding specs when spmd is enabled

JackCaoG committed 2 years ago

af494e21

JackCaoG committed 2 years ago

bc374c3d

Handle the different input for dynamo sharding change

JackCaoG committed 2 years ago

df2e27b1

JackCaoG committed 2 years ago

dbca95e6

Support input sharding changed after first dynamo tracing

JackCaoG committed 2 years ago

3412fe8f

Add topological sorting to dynamo partitions (#5472)

wonjoo-wj committed 2 years ago

Verified b58327e1

Add ShardingSpec to XLATensor when it is created with a PJRTShardedData (#5489)

JackCaoG committed 2 years ago

Verified 6270cbae

Update artifacts.auto.tfvars for 2.1 release (#5483)

wonjoo-wj committed 2 years ago

Verified 014caa53

[Dist] Refactor ZeRO-1 (#5145)

hgt312 committed 2 years ago

Verified 059f6b9d

[SPMD] Fix XLA_DUMP_POST_OPTIMIZATIONS test (#5485)

alanwaketan committed 2 years ago

Verified f1e1e271

[SPMD] Fix TPU CI after #5478 (#5487)

alanwaketan committed 2 years ago

Verified 8a98a887

Add Python API to execute StableHLO bytecode (#5476)

lsy323 committed 2 years ago

Verified a2fab36f

Add @yeounoh to .github CODEOWNERS (#5482)

yeounoh committed 2 years ago

Verified 2d0a729f

[SPMD] Properly skip tests on TPU V2 (#5479)

alanwaketan committed 2 years ago

Verified 364819a9

[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

Older