xla
[SPMD] Support mark_shard scalar tensors
#6158
Merged

[SPMD] Support mark_shard scalar tensors #6158

alanwaketan merged 1 commit into r2.2 from alanwaketan/scalar-backport
alanwaketan
alanwaketan alanwaketan requested a review from yeounoh yeounoh 1 year ago
alanwaketan alanwaketan requested a review from zpcore zpcore 1 year ago
alanwaketan alanwaketan requested a review from jonb377 jonb377 1 year ago
alanwaketan alanwaketan requested a review from JackCaoG JackCaoG 1 year ago
alanwaketan alanwaketan assigned alanwaketan alanwaketan 1 year ago
jonb377
jonb377 approved these changes on 2023-12-14
alanwaketan [SPMD] Add a test case for mark_shard scalar tensors
fde87430
alanwaketan alanwaketan force pushed from dc58b2eb to fde87430 1 year ago
alanwaketan
alanwaketan alanwaketan merged 98df3d1a into r2.2 1 year ago
alanwaketan alanwaketan changed the title [SPMD] Add a test case for mark_shard scalar tensors [SPMD] Support mark_shard scalar tensors 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone