xla
Pin all layout by default and use all_reduce based all_gather when la…
#3568

Merged

Pin all layout by default and use all_reduce based all_gather when la… #3568

JackCaoG merged 3 commits into master from all_gather_use_all_reduce

Pin all layout by default and use all_reduce based all_gather when la…

99c5c981

JackCaoG requested a review from

miladm 3 years ago

JackCaoG requested a review from

hjm-aws 3 years ago

hjm-aws commented on 2022-05-16

Only using all_reduce to implement all_gather for TPU and GPU

197a2de5

fix typo

0817a64d

JackCaoG requested a review from

hjm-aws 3 years ago

hjm-aws approved these changes on 2022-05-18

JackCaoG merged 7f96fbf6 into master 3 years ago

JackCaoG deleted the all_gather_use_all_reduce branch 3 years ago

Reviewers

hjm-aws

miladm

Assignees

No one assigned

Labels

None yet

Milestone

No milestone