xla
Fix xm.all_gather for non-empty groups
#3413
Merged

Fix xm.all_gather for non-empty groups #3413

yeounoh merged 2 commits into master from fix_xm_all_gather
yeounoh
yeounoh yeounoh added DO_NOT_MERGE_YET
yeounoh yeounoh requested a review from JackCaoG JackCaoG 4 years ago
yeounoh yeounoh requested a review from hjm-aws hjm-aws 4 years ago
yeounoh
yeounoh yeounoh marked this pull request as draft 4 years ago
yeounoh yeounoh force pushed from 1ea36199 to 2e0777d8 4 years ago
yeounoh yeounoh force pushed from 2e0777d8 to 4322dc63 4 years ago
yeounoh yeounoh force pushed from 4322dc63 to c392c62a 4 years ago
yeounoh yeounoh force pushed from c392c62a to e5d8cff3 4 years ago
yeounoh yeounoh force pushed from e5d8cff3 to 84954376 4 years ago
yeounoh yeounoh force pushed from 84954376 to 6bfd428e 4 years ago
yeounoh yeounoh force pushed from 6bfd428e to 06dcfaf7 4 years ago
yeounoh yeounoh force pushed from 06dcfaf7 to 43918924 4 years ago
yeounoh yeounoh force pushed from 43918924 to c8c95aa6 4 years ago
yeounoh yeounoh force pushed from c8c95aa6 to 0ef8f5b4 4 years ago
yeounoh yeounoh force pushed from 0ef8f5b4 to 5e153a83 4 years ago
yeounoh yeounoh marked this pull request as ready for review 4 years ago
yeounoh yeounoh removed DO_NOT_MERGE_YET
yeounoh
yeounoh yeounoh changed the title Fix xm all gather Fix xm.all_gather 4 years ago
yeounoh yeounoh changed the title Fix xm.all_gather Fix xm.all_gather for non-empty groups 4 years ago
JackCaoG
JackCaoG commented on 2022-03-08
JackCaoG
JackCaoG approved these changes on 2022-03-08
hjm-aws
hjm-aws approved these changes on 2022-03-08
yeounoh yeounoh force pushed from 5e153a83 to 12b7cea7 4 years ago
yeounoh Set proper shard_count for all_gather, when replica groups are non-em…
7cdc1f2c
yeounoh Update test_mp_all_gather.py
278fdaf5
yeounoh yeounoh force pushed from 12b7cea7 to 278fdaf5 4 years ago
yeounoh yeounoh assigned yeounoh yeounoh 4 years ago
yeounoh yeounoh merged c74deb63 into master 4 years ago
yeounoh yeounoh deleted the fix_xm_all_gather branch 4 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone