xla
Support `dist.all_gather` related collective ops
#7860

Merged

Support `dist.all_gather` related collective ops #7860

zpcore merged 8 commits into master from piz/cop

zpcore force pushed from 3529d208 to ca0fd5d9 1 year ago

zpcore requested a review from

will-cromar 1 year ago

zpcore requested a review from

JackCaoG 1 year ago

zpcore force pushed from ca0fd5d9 to 6e23e8a6 1 year ago

zpcore added tpuci

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

zpcore requested a review from

lsy323 1 year ago

zpcore changed the title ~~prototype of all_gather related distributed ops~~ Support `dist.all_gather` related distributed ops 1 year ago

zpcore changed the title ~~Support `dist.all_gather` related distributed ops~~ Support `dist.all_gather` related collective ops 1 year ago

will-cromar commented on 2024-08-19

lsy323 removed review request from

lsy323 1 year ago

zpcore force pushed from 7ec8f124 to b37add25 1 year ago

zpcore requested a review from

will-cromar 1 year ago

will-cromar commented on 2024-08-21

zpcore requested a review from

will-cromar 1 year ago

will-cromar commented on 2024-08-22

zpcore requested a review from

will-cromar 1 year ago

zpcore force pushed from 09bba568 to 0dd525c6 1 year ago

will-cromar approved these changes on 2024-08-26

add test case

4b988c3e

nit

ca757f94

nit

15b80593

clean up test and add dist.all_gather

6e972957

remove mock object

65989e31

nit

583d984f

update to openxla backend

ecac71b9

use toy backend

dfe8a542

zpcore force pushed from 0dd525c6 to dfe8a542 1 year ago

zpcore merged f9a706e0 into master 1 year ago

zpcore deleted the piz/cop branch 1 year ago

Reviewers

will-cromar

JackCaoG

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

xla Support `dist.all_gather` related collective ops #7860 Merged

Support `dist.all_gather` related collective ops #7860

xla
Support `dist.all_gather` related collective ops
#7860

Merged