xla
Support input sharding changed after first dynamo tracing
#5477

Merged

Support input sharding changed after first dynamo tracing #5477

JackCaoG merged 9 commits into master from JackCaoG/spmd_dynamo_sharding_changed

JackCaoG added distributed

JackCaoG added dynamo

JackCaoG commented on 2023-08-22

JackCaoG force pushed from 8500415f to 75d42d1e 2 years ago

JackCaoG requested a review from

alanwaketan 2 years ago

JackCaoG requested a review from

yeounoh 2 years ago

JackCaoG requested a review from

wonjoo-wj 2 years ago

Support input sharding changed after first dynamo tracing

3412fe8f

fix linter

dbca95e6

Handle the different input for dynamo sharding change

df2e27b1

update counter

bc374c3d

only get sharding specs when spmd is enabled

af494e21

add option to skip checking input sharding after x runs

b9ec261b

handle the cpu test

cd29929f

make XLA_DYNAMO_INPUT_SHARDING_CHECK_THREASHOLD configable

ca235465

JackCaoG force pushed from bea41098 to ca235465 2 years ago

wonjoo-wj approved these changes on 2023-08-24

fix review comments

0f4417d7

alanwaketan commented on 2023-08-24

alanwaketan approved these changes on 2023-08-25

JackCaoG merged d1212e90 into master 2 years ago

Reviewers

alanwaketan

wonjoo-wj

yeounoh

Assignees

No one assigned

Labels

distributed dynamo

Milestone

No milestone