pytorch/xla

Pull Requests Commits

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

Switch to pytorch sccache (#4489)

huydhn committed 2 years ago

Verified 6bf36302

Remove deprecated torch.symeig (#4498)

IvanYashchuk committed 2 years ago

Verified 903f2ac5

Update the tf pin to 2023 01/23 (#4474)

JackCaoG committed 2 years ago

Verified 2baef35b

Don't use futures on output buffers (#4503)

will-cromar committed 2 years ago

Verified b5d6be81

Fix HLO dump file names for PJRT (#4487)

will-cromar committed 2 years ago

Verified b9efeeb4

Disable nonzero test to unblock the head (#4512)

JackCaoG committed 2 years ago

Verified 5714e03f

Disable more dynamic shape tests (#4507)

JackCaoG committed 2 years ago

Verified 05195c7e

Temporarily disable test TestDynamicShape.test_nonzero_cast (#4502)

vanbasten23 committed 2 years ago

Verified bc14fe02

refactor: move dynamo/TorchXLA bridge to pytorch/xla repo (#4476)

shunting314 committed 2 years ago

Verified 0c9801bc

Revert "Squashed commit of the following: (#3272)" (#4497)

JackCaoG committed 2 years ago

Verified 844f137c

Squashed commit of the following: (#3272)

IvanYashchuk committed 2 years ago

Verified 282567af

Newer Older