pytorch/xla

Pull Requests Commits

Update openxla.md

ManfeiBai committed 2 years ago

Verified 897d696b

Update openxla.md

ManfeiBai committed 2 years ago

Verified ab61ea23

Update openxla.md

ManfeiBai committed 2 years ago

Verified f04aa815

Update openxla.md

ManfeiBai committed 2 years ago

Verified 4ebc7a40

Update openxla.md

ManfeiBai committed 2 years ago

Verified abd32e04

Update openxla.md

ManfeiBai committed 2 years ago

Verified 27bef8c5

Update openxla.md

ManfeiBai committed 2 years ago

Verified 7188db17

Update openxla.md

ManfeiBai committed 2 years ago

Verified df52fb49

Update openxla.md

ManfeiBai committed 2 years ago

Verified faa392fd

Create openxla.md

ManfeiBai committed 2 years ago

Verified b8e0c1f9

Fix tupled partition spec test on v3 (#5503)

jonb377 committed 2 years ago

Verified 81063fb8

remove torchvision in stablehlo.py (#5501)

lsy323 committed 2 years ago

Verified 1c42ba74

Add explict error message when tensor is on CPU for dynamo backend (#5499)

JackCaoG committed 2 years ago

Verified 9557fb72

Add a API to explictly init runtime (#5500)

JackCaoG committed 2 years ago

Verified e1ee592d

Support tuples in partition spec (#5488)

jonb377 committed 2 years ago

Verified 7a63c276

Refactor stablehlo API and put them in official location. (#5493)

qihqi committed 2 years ago

Verified 17abce76

Skip a couple tests on TPU due to precision issue (#5496)

JackCaoG committed 2 years ago

Verified d3d6ca30

Always use ExecuteReplicated with SPMD (#5494)

jonb377 committed 2 years ago

Verified 03991d44

Support input sharding changed after first dynamo tracing (#5477)

JackCaoG committed 2 years ago

Verified d1212e90

[SPMD] Make IR sharding custom sharding op (#5433)

alanwaketan committed 2 years ago

Verified dea9b296

[original author: mrnikwaws] Neuron operator support (#5471)

aws-kingrj committed 2 years ago

Verified 3e370b50

[SPMD] Patch nn.Linear (#5491)

alanwaketan committed 2 years ago

Verified 4f378e69

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

Older