xla
[SPMD] auto-construct auto-sharding mesh ids
#6770

Merged

[SPMD] auto-construct auto-sharding mesh ids #6770

yeounoh merged 4 commits into master from auto_spmd_mesh_ids

yeounoh added distributed

yeounoh requested a review from

JackCaoG 2 years ago

yeounoh assigned

yeounoh 2 years ago

JackCaoG added backport_2.3

JackCaoG commented on 2024-03-19

yeounoh force pushed from d8785ea2 to 2837c53e 2 years ago

yeounoh commented on 2024-03-19

yeounoh force pushed from 339222b9 to 0a66c90b 2 years ago

yeounoh commented on 2024-03-19

yeounoh requested a review from

lsy323 2 years ago

yeounoh requested a review from

zpcore 2 years ago

lsy323 approved these changes on 2024-03-19

JackCaoG approved these changes on 2024-03-19

yeounoh added DO_NOT_MERGE_YET

yeounoh force pushed from 0a66c90b to ada802e3 2 years ago

[SPMD] construct auto-sharding mesh ids based on the buitl XLA comput…

3496de89

Move openxla pin to the HEAD

f7d8345c

Delete quant_dequant_converter.diff, since it is part of openxla now

423fda06

Move to 20240320 nightly

ff7818cf

yeounoh force pushed from ada802e3 to ff7818cf 2 years ago

yeounoh removed DO_NOT_MERGE_YET

yeounoh merged 968ac834 into master 2 years ago

Reviewers

JackCaoG

lsy323

zpcore

Assignees

yeounoh

Labels

distributed backport_2.3

Milestone

No milestone