PR #5301 [SPMD] Support mark_sharding on IRs

[SPMD] Support mark_sharding on IRs #5301

alanwaketan merged 5 commits into master from alanwaketan/spmd_ir

[SPMD] Avoid recompilations in xs.mark_sharding()

eb151429

[SPMD] Support mark_sharding on IRs

938d5fb0

Fix rebase error

632c9fea

alanwaketan requested a review from

jonb377 2 years ago

alanwaketan requested a review from

JackCaoG 2 years ago

alanwaketan requested a review from

khatwanimohit 2 years ago

Fix linters

d22e642b

Unskip test_2d_tensor_3d_mesh

bab326ec

JackCaoG approved these changes on 2023-07-13

alanwaketan merged 6fe5cb98 into master 2 years ago

Reviewers

JackCaoG

jonb377

khatwanimohit

Assignees

No one assigned

Labels

None yet

Milestone

No milestone