pytorch/xla

Pull Requests Commits

vanbasten23 committed 2 years ago

8b1ce3c4

Test on GPU is skipped. Fixed it.

vanbasten23 committed 2 years ago

3ee26870

improve the test

vanbasten23 committed 2 years ago

3ab082ed

make the test show which speicifc test fails via Verbose flag.

vanbasten23 committed 2 years ago

236c5243

should expect the test fail.

vanbasten23 committed 2 years ago

585c2d8c

vanbasten23 committed 2 years ago

57c14e95

Revert "disable the test on CPU."

vanbasten23 committed 2 years ago

75b22e6f

Revert "fix linter"

vanbasten23 committed 2 years ago

a63054d8

vanbasten23 committed 2 years ago

14e30deb

disable the test on CPU.

vanbasten23 committed 2 years ago

a49baf41

vanbasten23 committed 2 years ago

bd3363f8

vanbasten23 committed 2 years ago

7269590c

fix pr comments

vanbasten23 committed 2 years ago

300640cd

add verbose flag.

vanbasten23 committed 2 years ago

6a87404d

vanbasten23 committed 2 years ago

2717f76d

Add ds model test to the ci.

vanbasten23 committed 2 years ago

df8d18d8

Fix `non_zero` cast issue (#4243)

JackCaoG committed 2 years ago

Verified 48e12e3d

Update README to remove tpu-node wheels and add gpu 3.8 wheels and dockers (#4251)

JackCaoG committed 2 years ago

Verified d668f2ca

Experimental TPU implementation of DistributedDataParallel (#4193)

will-cromar committed 2 years ago

Verified 803052c1

Fix 1d transpose (#4246)

JackCaoG committed 2 years ago

Verified 0fff6b66

[LTC] Replace XLA_TIMED with TORCH_LAZY_TIMED (#4250)

alanwaketan committed 2 years ago

Verified fc51c9f2

CI requires cmake>=3.18 to install PyTorch (#4247)

yeounoh committed 2 years ago

Verified 519f5942

Try to make op tests runs on TPUVM (#4244)

JackCaoG committed 2 years ago

Verified bb089b0f

[LTC] Replace XLA_VALUE_METRIC with TORCH_LAZY_VALUE_METRIC (#4245)

alanwaketan committed 2 years ago

Verified 09498a85

Add comment to stream executor build file patch (#4249)

steventk-g committed 2 years ago

Verified e564f825

[LTC] Replace XLA_FN_COUNTER with TORCH_LAZY_FN_COUNTER (#4217)

alanwaketan committed 2 years ago

Verified b969cba3

Companion PR for pytorch/pytorch#89327 (#4242)

Nikita Karetnikov committed 2 years ago

Verified 71c5a005

[SPMD] fixes bugs with AssignIrValue & ExecuteReplicated (#4233)

yeounoh committed 2 years ago

Verified 25cb9975

Optionally bundle libtpu into TPU VM wheel (#4204)

will-cromar committed 2 years ago

Verified f8581559

Add fork start_method support for xmp.spawn (#4236)

darisoy committed 2 years ago

Verified 7c0acfc7

Older