xla
Add API to assemble CPU shards to a sharded tensor
#5630

Merged

Add API to assemble CPU shards to a sharded tensor #5630

jonb377 merged 4 commits into master from jonbolin-assemble-shards

jonb377 requested a review from

yeounoh 2 years ago

jonb377 requested a review from

alanwaketan 2 years ago

jonb377 requested a review from

JackCaoG 2 years ago

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

jonb377 force pushed from 34cc1d30 to a1e83ca2 2 years ago

Add API to assemble CPU shards to a sharded tensor

13cfb7ac

jonb377 force pushed from a1e83ca2 to 13cfb7ac 2 years ago

jonb377 commented on 2023-09-20

Handle replicated sharding

4413bc7a

jonb377 force pushed from 5b696f05 to 4413bc7a 2 years ago

Move validations into get_op_sharding

bf329c7b

jonb377 assigned

jonb377 2 years ago

jonb377 added distributed

yeounoh commented on 2023-09-22

alanwaketan commented on 2023-09-22

Improve tests and error handling

c03a5672

jonb377 force pushed from 7bd0555c to c03a5672 2 years ago

yeounoh approved these changes on 2023-10-05

alanwaketan approved these changes on 2023-10-05

jonb377 merged 3913a77e into master 2 years ago

jonb377 deleted the jonbolin-assemble-shards branch 2 years ago

Reviewers

alanwaketan

yeounoh

JackCaoG

Assignees

jonb377

Labels

distributed

Milestone

No milestone