pytorch/xla

Pull Requests Commits

alanwaketan committed 1 year ago

ed5254e3

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

Properly lower add and mul (#6731) (#6744)

lsy323 committed 1 year ago

Verified f9c94d9a

Properly lower aten::sub and aten::rsub (#6668) (#6743)

lsy323 committed 1 year ago

Verified 9cb18dfa

[Pallas] Introduce make_kernel_from_pallas (#6713) (#6742)

lsy323 committed 1 year ago

Verified e556ad81

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

lsy323 committed 1 year ago

Verified a47f81d9

Update `torch_xla_cuda_plugin` version for r2.3 (#6729)

will-cromar committed 1 year ago

Verified aa6d2775

add torch pin to 2.3 release branch (#6735)

lsy323 committed 1 year ago

Verified 45268bd1

[r2.3] backport #6723 (#6734)

yeounoh committed 1 year ago

Verified 085059c4

Backport #6730 (#6739)

lsy323 committed 1 year ago

Verified b70446a1

Backport #6677 to r2.3 (#6737)

lsy323 committed 1 year ago

Verified db30eb32

fix csv output error (#6662) (#6738)

zpcore committed 1 year ago

Verified a188f7e7

Backport #6704 to r2.3 (#6736)

lsy323 committed 1 year ago

Verified c5a7ff76

Add IsVirtualDeivce() and refactor

yeounoh committed 1 year ago

866c23be

Unbundle libtpu in CI (#6727)

will-cromar committed 1 year ago

Verified 21c1c63e

add tpu path in tpu ci run test (#6717)

lsy323 committed 1 year ago

Verified bf60b623

Allow `int` data-type for `Embedding` indices. (#6718)

ysiraichi committed 1 year ago

Verified 5a7bcac7

Enable unbounded dynamism on conv, softmax, addmm, slice (#6494)

lsy323 committed 1 year ago

Verified 97acc14c

[SPMD] Remove unused spmd partitioning debugging tool. (#6720)

yeounoh committed 1 year ago

Verified 5494a63b

Ignore .vscode/ and move settings.json as template (#6705)

yeounoh committed 1 year ago

Verified 92bb381a

Older