xla
[Distributed] Switch all_reduce to use the new functional collective op
#6887

Merged

[Distributed] Switch all_reduce to use the new functional collective op #6887

alanwaketan merged 1 commit into pytorch:master from yifuwang:all_reduce

yifuwang force pushed from b4ea4a54 to f20991ce 1 year ago

JackCaoG requested a review from

alanwaketan 1 year ago

yifuwang marked this pull request as draft 1 year ago

yifuwang force pushed from f20991ce to 51ff1576 1 year ago

alanwaketan commented on 2024-04-08

[Distributed] Switch all_reduce to use the new functional collective op

f3625aa3

yifuwang force pushed from 51ff1576 to f3625aa3 1 year ago

yifuwang marked this pull request as ready for review 1 year ago

alanwaketan approved these changes on 2024-04-10

alanwaketan merged a816c42f into master 1 year ago

Reviewers

alanwaketan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone