Add XLA backend for torch.distributed #3339
JackCaoG
merged
16 commits into
pytorch:master
from
xla_backend2
Add XLA backend for torch.distributed
f8510293
Add XLA backend multiprocessing tests.
e0d5292d
Linter fixes.
08d34536
Address Jack's comments.
6fe7c990
Fix multiprocessing tests: forgot to import the backend.
1f821d29
Addressing Shen and Jack's comments.
df3319aa
Fix a search/replace error.
5fb9b83b
Fix typo in test_mp_all_gather_xla_backend to make it real.
0bfe021f
Use new reduce_scatter output param and use tensor.copy_ for all_gath…
566703bd
Lint fix.
abb6e41e
Fix TODO(alias).
6f0c9f46
Add XRT_WORKERS and XRT_DEVICE_MAP setting back to the unit test as w…
cb221505
Lint fix.
ee3f553c
Skip XLA backend unit tests for GPU/TPU.
2b0bd721
Address Jack's comments.
c3827d76
mrshenli
approved these changes
on 2022-02-15
JackCaoG
requested changes
on 2022-02-15
rename tests according to Jack's comment.
fe0bf9d5
JackCaoG
approved these changes
on 2022-02-15
Login to write a write a comment.
Login via GitHub