pytorch/xla

Pull Requests Commits

Fix preserve_rng_state for activation checkpointing (#4690)

YangFei1990 committed 1 year ago

49efb2b6

[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

Install torch wheel from pytorch to unblock TPU CI (#6691)

vanbasten23 committed 1 year ago

Verified 6630287c

Disable pallas test in tpu ci (#6715)

lsy323 committed 1 year ago

Verified 16617828

Add more dtypes; add jax scope for debugging (#6714)

qihqi committed 1 year ago

Verified 707e8fb3

[SPMD] Rename conflicting test class names (#6712)

yeounoh committed 1 year ago

Verified 5c4ed89b

Remove `torchvision` from `test_operations` (#6698)

will-cromar committed 1 year ago

Verified c0509fc6

Don't terminate the whole process when `Compile` fails (#6707)

will-cromar committed 1 year ago

Verified d503b71e

[Pallas] Support programmatically extracting the payload (#6696)

alanwaketan committed 1 year ago

Verified 37067917

Older