pytorch/xla

Pull Requests Commits

Fix preserve_rng_state for activation checkpointing (#4690)

YangFei1990 committed 1 year ago

Verified 66acfebf

Remove extraneous cd (#6787)

mlevesquedion committed 1 year ago

Verified 2d0f91c7

Add dynamism support to conv1d, view, softmax (#6777)

lsy323 committed 1 year ago

Verified e9581bc9

[SPMD] auto-construct auto-sharding mesh ids (#6770)

yeounoh committed 1 year ago

Verified 968ac834

Address deprecation in torch.distributed.checkpoint (#6773)

jonb377 committed 1 year ago

Verified 73629de2

Move multislice sharding example into torch_xla (#6776)

jonb377 committed 1 year ago

Verified f1c61d32

Add fx passes to support exporting unbounded dynamism (#6653)

lsy323 committed 1 year ago

Verified 7e0d3a5a

[Pallas] Add a bfloat16 flash attention test case (#6748)

alanwaketan committed 1 year ago

Verified 1ccd6a64

[Pallas] Introduce jax_import_guard (#6747)

alanwaketan committed 1 year ago

Verified 7cf9f10a

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

Newer Older