pytorch/xla

Pull Requests Commits

comment out the test

vanbasten23 committed 1 year ago

0101d7be

use pytorch profiler

vanbasten23 committed 1 year ago

dc48e9ba

Properly lower add and mul (#6731)

wonjoo-wj committed 1 year ago

Verified fe3f23c6

Add Build Trigger for 2.3-rc1 release (#6688)

lsy323 committed 1 year ago

Verified 548942de

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

will-cromar committed 1 year ago

Verified 977e7efc

[Pallas] Introduce make_kernel_from_pallas (#6713)

alanwaketan committed 1 year ago

Verified 1bbe333a

[SPMD] Use UNKNOWN sharding type for implicit replication (#6723)

yeounoh committed 1 year ago

Verified cc55d9e7

install pinned torchvision for Github Action TPU CI (#6730)

lsy323 committed 1 year ago

Verified e8580e77

Pin update March 2024 (#6677)

lsy323 committed 1 year ago

Verified 82b5ed3a

Automatically move CUDA non XLA Tensors to XLA Device and back to CUDA device (#6644)

changm committed 1 year ago

Verified d13ae1bb

Add IsVirtualDeivce() and refactor (#6726)

yeounoh committed 1 year ago

Verified 7d899138

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

anw90 committed 1 year ago

Verified 50ae882a

fix csv output error (#6662)

zpcore committed 1 year ago

Verified 024b576b

fix torchvision installation in tpu ci test setup script (#6704)

lsy323 committed 1 year ago

Verified d5a079ed

Skip the execution if all Pending IRs are device data (#6642)

JackCaoG committed 1 year ago

Verified 70962202

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

ezyang committed 1 year ago

Verified 84419438

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

Older