pytorch/xla

Pull Requests Commits

alanwaketan committed 1 year ago

0115d0fe

alanwaketan committed 1 year ago

a82b5c6e

Guard JAX import for TPU tests only

alanwaketan committed 1 year ago

e42b9727

Move JAX import into make_kernel_from_pallas and use _init_computation_client

alanwaketan committed 1 year ago

d127022f

alanwaketan committed 1 year ago

fe143834

alanwaketan committed 1 year ago

f694d02c

alanwaketan committed 1 year ago

ce2130e5

Revert "Reland "Removing nondeterminism from partitioner eliminates one graph" (#6703)" (#6774)

JackCaoG committed 1 year ago

Verified 30a261ab

Skip `cdouble` dtype in pow test on TPU (#6772)

lsy323 committed 1 year ago

Verified bd4420c5

Fallback to CPU for dot implementation for long tensors (#6708)

bhavya01 committed 1 year ago

Verified 7145a719

Re-land: Make `as_strided_copy` materialize a new tensor with `index`. (#6697)

ysiraichi committed 1 year ago

Verified 27a7dd35

Add unit tests for a few `foo.Tensor` aten ops (#6749)

ghpvnist committed 1 year ago

Verified 6ac32233

Remove ancient setup script (#6692)

will-cromar committed 1 year ago

Verified ff2bfe66

Fix type promotion for `pow`. (#6745)

ysiraichi committed 1 year ago

Verified 902aa506

Fix torchrun for single-process case. (#6764)

vanbasten23 committed 1 year ago

Verified 790e5c80

remove openxla_eval from the doc (#6757)

JackCaoG committed 1 year ago

Verified d205f9d0

Disable coverage temporarily (#6746)

vanbasten23 committed 1 year ago

Verified f04870e4

Include auto-sharding configs as part of hash (#6753)

yeounoh committed 1 year ago

Verified dbe2bc2a

Build torchvision from the pytorch pin in ansible (#6756)

will-cromar committed 1 year ago

Verified b8ffb452

Properly lower div.scalar and div.tensor (#6669)

bhavya01 committed 1 year ago

Verified 19b83830

[SPMD] auto-sharding PoC (#6719)

yeounoh committed 1 year ago

Verified 370089a9

Properly lower add and mul (#6731)

wonjoo-wj committed 1 year ago

Verified fe3f23c6

Add Build Trigger for 2.3-rc1 release (#6688)

lsy323 committed 1 year ago

Verified 548942de

Bump JAX version in `setup.py` (#6740)

will-cromar committed 1 year ago

Verified 977e7efc

[Pallas] Introduce make_kernel_from_pallas (#6713)

alanwaketan committed 1 year ago

Verified 1bbe333a

[SPMD] Use UNKNOWN sharding type for implicit replication (#6723)

yeounoh committed 1 year ago

Verified cc55d9e7

install pinned torchvision for Github Action TPU CI (#6730)

lsy323 committed 1 year ago

Verified e8580e77

Pin update March 2024 (#6677)

lsy323 committed 1 year ago

Verified 82b5ed3a

Automatically move CUDA non XLA Tensors to XLA Device and back to CUDA device (#6644)

changm committed 1 year ago

Verified d13ae1bb

Add IsVirtualDeivce() and refactor (#6726)

yeounoh committed 1 year ago

Verified 7d899138

Older