xla
[SPMD] Support manual all-reduce
#7576
Merged

[SPMD] Support manual all-reduce #7576

alanwaketan merged 2 commits into master from alanwaketan/spmd_all_reduce
alanwaketan
alanwaketan initiial commit
d3795b35
alanwaketan Fix linters
7f7739a5
alanwaketan alanwaketan added backport_2.4
alanwaketan alanwaketan requested a review from jonb377 jonb377 1 year ago
alanwaketan alanwaketan requested a review from JackCaoG JackCaoG 1 year ago
alanwaketan alanwaketan assigned alanwaketan alanwaketan 1 year ago
JackCaoG
JackCaoG approved these changes on 2024-06-26
JackCaoG
alanwaketan
alanwaketan
alanwaketan alanwaketan merged 0df5c291 into master 1 year ago
alanwaketan alanwaketan deleted the alanwaketan/spmd_all_reduce branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone