pytorch/xla

Pull Requests Commits

Update test_fori_loop_with_while_loop_simple_add_dispatch_in_torch.py

ManfeiBai committed 1 year ago

Verified 40796dfe

add nested test case for fori_loop

ManfeiBai committed 1 year ago

Verified 29af77df

Add cuda 12.1 nightly wheels for cp310 and cp311. (#6793)

vanbasten23 committed 1 year ago

Verified 7d8972c8

ZeRO1: Add bucketting logic to control the size of tensors for all-gather/reduce-scatter (#6025)

jeffhataws committed 1 year ago

Verified e75677f1

Update the instruction of installing the NVIDIA container toolkit (#6791)

vanbasten23 committed 1 year ago

Verified 782f05d2

[Pallas] Allow setting FlashAttention's causal mask (#6792)

alanwaketan committed 1 year ago

Verified de42834d

Add patch for unbounded dynamism (#6783)

ghpvnist committed 1 year ago

Verified 8d579f91

Use stablehlo.composite instead of stablehlo.custom_call (#6789)

mlevesquedion committed 1 year ago

Verified 3eeb15d9

Use AddBufferDonor to replace SetUpAlias for input output aliasing (#6769)

JackCaoG committed 1 year ago

Verified d70b6c4e

Add dynamism support to `aten.embedding` and `aten.split_with_sizes` (#6781)

lsy323 committed 1 year ago

Verified 6b76c7f6

[Pallas] Make a FlashAttention Wrapper (#6785)

alanwaketan committed 1 year ago

Verified fcf24b6c

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

Older