xla
[SPMD] Avoid recompilations in xs.mark_sharding()
#5300

Merged

[SPMD] Avoid recompilations in xs.mark_sharding() #5300

alanwaketan merged 4 commits into master from alanwaketan/spmd_mark

alanwaketan requested a review from

jonb377 2 years ago

alanwaketan requested a review from

JackCaoG 2 years ago

alanwaketan assigned

alanwaketan 2 years ago

JackCaoG commented on 2023-07-12

alanwaketan requested a review from

khatwanimohit 2 years ago

[SPMD] Avoid recompilations in xs.mark_sharding()

2b02b7b7

Fix linters

d1619610

Skip test_2d_tensor_3d_mesh given it relies on IR sharding

326ce26c

alanwaketan force pushed from c3774970 to 326ce26c 2 years ago

Address comments

55de0096

JackCaoG commented on 2023-07-13

JackCaoG approved these changes on 2023-07-13

alanwaketan merged caf51687 into master 2 years ago

Reviewers

JackCaoG

jonb377

khatwanimohit

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone