xla
[SPMD] Support SPMDFullToShardShape
#6922
Merged
Overview
Commits
14
Changes
View On GitHub

[SPMD] Support SPMDFullToShardShape #6922

alanwaketan merged 14 commits into master from alanwaketan/pause_spmd
alanwaketan
alanwaketan initial commit
74b494df
alanwaketan Fix linters
fa34ef68
alanwaketan Update comment
0e0ad181
alanwaketan Disallow mark manual sharding on data tensors
a1120519
alanwaketan initial commit for pause
b3ee21a1
alanwaketan expand custom sharding
cfa9c560
alanwaketan Make it even more extensible
a6d1be2b
alanwaketan Make test case and fix it
3ececd34
alanwaketan Add more tests
d77f85c9
alanwaketan Fix linters
ef834bf4
alanwaketan alanwaketan requested a review from yeounoh yeounoh 1 year ago
alanwaketan alanwaketan requested a review from jonb377 jonb377 1 year ago
alanwaketan Fix rebase error
d9c3e24c
JackCaoG
JackCaoG commented on 2024-04-12
alanwaketan address comments
52bf2901
alanwaketan Make kSharding as default
c70900b2
alanwaketan Fix linters
7098b3da
yeounoh
yeounoh approved these changes on 2024-04-15
alanwaketan
alanwaketan alanwaketan merged 2763248f into master 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum