pytorch/xla

Pull Requests Commits

fix a failing test.

vanbasten23 committed 2 years ago

f7124d3f

vanbasten23 committed 2 years ago

2b446462

unify the op kind

vanbasten23 committed 2 years ago

af7f1d8f

vanbasten23 committed 2 years ago

41eab204

check counter in the test.

vanbasten23 committed 2 years ago

51a8ed5a

change opkind to be more specific

vanbasten23 committed 2 years ago

10919c8d

vanbasten23 committed 2 years ago

8ae3ac66

the code buillds fine and the test passed.

vanbasten23 committed 2 years ago

510ae375

implementing xlasymnodeimpl::sub

vanbasten23 committed 2 years ago

d14ee8e1

make empty_symint support dynamism. (#4550)

vanbasten23 committed 2 years ago

Verified 34e91153

Reenable test_view_copy_out after https://github.com/pytorch/pytorch/pull/93411 (#4547)

yeounoh committed 2 years ago

Verified 4f05e6bb

Add SPMD benchmarking test (#4141)

jonb377 committed 2 years ago

Verified f5232da4

Reset Nvidia cuda-keyring in the CI env setup (#4548)

yeounoh committed 2 years ago

Verified 889a1039

Lower _linalg_svd (#4537)

wonjoo-wj committed 2 years ago

Verified b205042b

Transfer shards from server using replication (#4508)

yeounoh committed 2 years ago

Verified 780a8231

Align the dynamo dump file with the lazy (#4530)

JackCaoG committed 2 years ago

Verified ffd73b0d

Fix race condition between execution and buffer tear down on GPU when using bfc_allocator (#4542)

ymwangg committed 2 years ago

Verified 33e87a2f

Disable test_view_copy_out_xla (#4540)

yeounoh committed 2 years ago

Verified 9cbcdb40

Remove Python 3.7 build trigger from Terraform (#4534)

will-cromar committed 2 years ago

Verified ce48e42a

Optimize Dynamo Training Execute (#4523)

JackCaoG committed 2 years ago

Verified f5494cd7

Basic test for `PjrtComputationClient`. (#4478)

stgpetrovic committed 2 years ago

Verified c80cf439

[PJRT] Experimental support for `torch.distributed` and DDP on TPU v2/v3 (#4520)

will-cromar committed 2 years ago

Verified 021a1cc2

Enable python 3.8 and resolve dependency issues (#4529)

yeounoh committed 2 years ago

Verified bfd20f37

Try bootstrapping tpu env from env vars (#4499)

steventk-g committed 2 years ago

Verified f2cb76f9

Move XLA testing to Python-3.8 (#4527)

malfet committed 2 years ago

Verified 4f4ab8a3

Increase the timeout to make cuda docker build more stable (#4522)

JackCaoG committed 2 years ago

Verified b6b4da0d

[PJRT] Release the GIL during `TransferFromServer` (#4504)

will-cromar committed 2 years ago

Verified d41c5ba2

Always use `ExecuteSharded` future with PJRT C API. (#4521)

will-cromar committed 2 years ago

Verified efb53533

Disable the Sandbox build in Docker (#4515)

JackCaoG committed 2 years ago

Verified 661ab6d2

Bump up the xla_base image ver to v0.8 in upstream repo (#4516)

yeounoh committed 2 years ago

Verified 6acc0c9c

Older