xla
[Distributed] Remove sea-of-devices mode
#4905
Merged

[Distributed] Remove sea-of-devices mode #4905

alanwaketan merged 9 commits into master from alanwaketan/cc
alanwaketan
alanwaketan Remove sea of devices in all_reduce
5da034e4
alanwaketan Remove interhost allreduce
0d217d19
alanwaketan Remove _torch_all_reduce
fc9866f7
alanwaketan Remove CollectiveContext in reduce_gradients
8322f08e
alanwaketan Remove CollectiveContext
9d845b77
alanwaketan Further remove sea of devices
ea3088e4
alanwaketan alanwaketan requested a review from JackCaoG JackCaoG 2 years ago
alanwaketan alanwaketan assigned alanwaketan alanwaketan 2 years ago
alanwaketan Clean up even more
2ee39008
JackCaoG
JackCaoG approved these changes on 2023-04-19
alanwaketan
alanwaketan Fix linters
f4cb6ac7
alanwaketan Fix XlaBackendTest
721dd362
alanwaketan alanwaketan merged 397ea4d5 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone