pytorch/xla

Pull Requests Commits

alanwaketan committed 1 year ago

38054637

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

lsy323 committed 1 year ago

Verified b94c014e

Update 2.3 readme (#6957)

lsy323 committed 1 year ago

Verified 8c339793

Reland Install torchvision in release docker #6795 (#6860) (#6955)

lsy323 committed 1 year ago

Verified 76740307

[cherry-pick] [Fori_loop|While_loop] Create fori_loop.md (#6942) (#6945)

ManfeiBai committed 1 year ago

Verified 2ebd636b

Fix warning message when running mp workload (#6892)

JackCaoG committed 1 year ago

Verified 954242f6

[Backport] Add flash attention as custom op so dynamo can trace it (#6875) (#6884)

JackCaoG committed 1 year ago

Verified 96cc98ff

[Backport] Remove reshape lowering (#6801) (#6847)

wonjoo-wj committed 1 year ago

Verified 6e4b2ed4

[r2.3] Fix div lowering and core aten test script enhancement (#6873) (#6877)

lsy323 committed 1 year ago

Verified fabf9778

[Backport] Set the major-minor layout for inputs and outputs (#6876)

alanwaketan committed 1 year ago

Verified 3905372e

[r2,3][SPMD] use UNKNOWN as default sharded data sharding type selectively (#6852)

yeounoh committed 1 year ago

Verified c3d57654

[r2.3] Run pre-submit, post-submit TPU CI on the relesae branch (#6856) (#6859)

lsy323 committed 1 year ago

Verified 0d1f1723

[r2.3] Use AddBufferDonor to replace SetUpAlias for input output aliasing (#6855)

lsy323 committed 1 year ago

Verified 13a2169f

udpate release tag to rc6 (#6839)

lsy323 committed 1 year ago

Verified 235864a0

Make r2.3 to be compatible with PyPI (#6838)

lsy323 committed 1 year ago

Verified 1baea32d

[Backport] Support XLA_USE_BF16 (#6841)

alanwaketan committed 1 year ago

Verified c6a88740

[Backport] Allow setting FlashAttention's causal mask (#6837)

alanwaketan committed 1 year ago

Verified 046f9103

Document the CUDA plugin in r2.3 README (#6825)

will-cromar committed 1 year ago

Verified 8de8cea3

[Backport] Make a FlashAttention Wrapper (#6827)

alanwaketan committed 1 year ago

Verified db7112af

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

JackCaoG committed 1 year ago

Verified a805505d

[r2.3] Update XLA pin (#6797) (#6811)

yeounoh committed 1 year ago

Verified 7e8b8fb7

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

alanwaketan committed 1 year ago

Verified 3522be16

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

bhavya01 committed 1 year ago

Verified 8240d05b

Backport Fix preserve_rng_state for activation checkpointing (#4690) (#6788)

JackCaoG committed 1 year ago

Verified 1d6558a3

[Backport] Introduce jax_import_guard (#6794)

alanwaketan committed 1 year ago

Verified c3001d64

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

jonb377 committed 1 year ago

Verified d9877757

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

yeounoh committed 1 year ago

Verified c923e8f1

backport `remove openxla_eval from the doc` (#6767)

JackCaoG committed 1 year ago

Verified ebedf4d9

[r2.3] Include auto-sharding configs as part of hash (#6753) (#6763)

yeounoh committed 1 year ago

Verified a78586f8

[r2.3] backport: auto-sharding PoC (#6719) (#6755)

yeounoh committed 1 year ago

Verified 680ebc10

Older