PR #6915 [SPMD] Support manual sharding

[SPMD] Support manual sharding #6915

alanwaketan merged 7 commits into master from alanwaketan/spmd_manual

initial commit

d77e011c

Fix cpu test

39008eb0

Fix linters

e666e3b5

Update comment

995af75b

alanwaketan requested a review from

yeounoh 1 year ago

alanwaketan requested a review from

jonb377 1 year ago

alanwaketan assigned

alanwaketan 1 year ago

jonb377 commented on 2024-04-10

yeounoh commented on 2024-04-11

yeounoh approved these changes on 2024-04-11

yeounoh commented on 2024-04-11

Fix one comment

2db7f0a9

Disallow mark manual sharding on data tensors

84cd609b

Fix linters

faeb7c79

alanwaketan merged e5513ff4 into master 1 year ago

Reviewers

yeounoh

jonb377

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone