xla
[Distributed] Make xm.all_gather a single graph in Dynamo
#4922

Merged

[Distributed] Make xm.all_gather a single graph in Dynamo #4922

alanwaketan merged 10 commits into master from alanwaketan/all_gather

alanwaketan changed the base branch from master to alanwaketan/all_reduce_d 3 years ago

alanwaketan requested a review from

JackCaoG 3 years ago

alanwaketan requested a review from

wconstab 3 years ago

alanwaketan commented on 2023-04-21

alanwaketan force-pushed the alanwaketan/all_reduce_d branch from 7673dd6c to 89f7c6c2 3 years ago

alanwaketan force pushed from 86813851 to 674e53c6 3 years ago

alanwaketan changed the base branch from alanwaketan/all_reduce_d to alanwaketan/all_reduce_t 3 years ago

JackCaoG commented on 2023-04-21

JackCaoG approved these changes on 2023-04-21

Make all_gather single dynamo graph

0272c701

Fix a typo

83f6a16f

Fix V3-8

3aa4ebbf

Fix linters

a432f999

Cache world_size and ordinal in pjrt

f6ff59f8

skip tpu < v4 support

e1629cc5

Fix linters

da01f7f0

Fix comments

28a2a6eb

Rearrange the test

b3b4bd5f

Fix a typo

26cfb004

alanwaketan force pushed from 3fb46da1 to 26cfb004 3 years ago

alanwaketan changed the base branch from alanwaketan/all_reduce_t to master 3 years ago

alanwaketan merged 486b32e5 into master 3 years ago

Reviewers

JackCaoG

will-cromar

wconstab

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

xla [Distributed] Make xm.all_gather a single graph in Dynamo #4922 Merged

[Distributed] Make xm.all_gather a single graph in Dynamo #4922

xla
[Distributed] Make xm.all_gather a single graph in Dynamo
#4922

Merged