pytorch/xla

Pull Requests Commits

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

Ops refactor 1 (#6671)

qihqi committed 1 year ago

Verified 60431850

Revert "Removing nondeterminism from partitioner eliminates one graph" (#6703)

JackCaoG committed 1 year ago

Verified 707a6329

Make use_spmd to force SPMD mode for easier 3rd-party integration. (#6680)

yeounoh committed 1 year ago

Verified f3a2e243

[Fori_loop|While_loop] Enable fori_loop with add/sub test case (#6603)

ManfeiBai committed 1 year ago

Verified 6170df58

Revert "Change ATEN generator argument type to const std::optional<Generator>&" (#6701)

alanwaketan committed 1 year ago

Verified 6cc5c381

Not building cuda plugins to unblock upstream ci. (#6699)

vanbasten23 committed 1 year ago

Verified e99ab496

Removing nondeterminism from partitioner eliminates one graph (#6675)

ezyang committed 1 year ago

Verified d28053f9

Run test gpu device detection without setting PJRT_DEVICE (#6650)

changm committed 1 year ago

Verified 57f4780d

Newer Older