xla
[SPMD] cross slice-replication using partial_replication sharding
#5411

Merged

[SPMD] cross slice-replication using partial_replication sharding #5411

yeounoh merged 7 commits into master from spmd_cross_slice_replication

yeounoh assigned

yeounoh 2 years ago

yeounoh marked this pull request as draft 2 years ago

yeounoh marked this pull request as ready for review 2 years ago

yeounoh requested a review from

khatwanimohit 2 years ago

yeounoh requested a review from

JackCaoG 2 years ago

yeounoh requested a review from

jonb377 2 years ago

JackCaoG commented on 2023-08-07

jonb377 commented on 2023-08-07

Revert "Support unordered sharding spec for partial replication (#5316)"

8c0f3274

update test_2d_tensor_3d_mesh unit test to surface a bug

51f8e027

Use partial replication for 2D tensor over 3D mesh sharding

a3fa888a

JackCaoG commented on 2023-08-07

yeounoh force pushed from e82931f1 to 3eda28e5 2 years ago

Refactor test cases

78e11bb8

yeounoh force pushed from 3eda28e5 to 78e11bb8 2 years ago

Add comments

0af22044

yeounoh commented on 2023-08-07

Fix test helper code error

1beb120c

yeounoh force pushed from f5b084ea to 1beb120c 2 years ago

JackCaoG approved these changes on 2023-08-07

Linter fix

2b2bce8d

yeounoh commented on 2023-08-07

jonb377 approved these changes on 2023-08-07

alanwaketan approved these changes on 2023-08-07

khatwanimohit approved these changes on 2023-08-07

yeounoh merged 6852df1d into master 2 years ago

yeounoh changed the title ~~SPMD cross slice-replication using partial_replication sharding~~ [SPMD] cross slice-replication using partial_replication sharding 2 years ago

yeounoh added distributed

Reviewers

khatwanimohit

alanwaketan

jonb377

JackCaoG

Assignees

yeounoh

Labels

distributed

Milestone

No milestone

xla [SPMD] cross slice-replication using partial_replication sharding #5411 Merged

[SPMD] cross slice-replication using partial_replication sharding #5411

xla
[SPMD] cross slice-replication using partial_replication sharding
#5411

Merged