PR #6101 [SPMD] Allow mark_shard scalar tensors

[SPMD] Allow mark_shard scalar tensors #6101

alanwaketan merged 3 commits into master from alanwaketan/scalar

alanwaketan requested a review from

yeounoh 2 years ago

alanwaketan requested a review from

jonb377 2 years ago

alanwaketan assigned

alanwaketan 2 years ago

alanwaketan force pushed from 7622e24c to e65ed1d0 2 years ago

alanwaketan changed the title ~~[SPMD] Add a test case for mark_shard scalar tensors~~ [SPMD] Allow mark_shard scalar tensors 2 years ago

alanwaketan requested a review from

JackCaoG 2 years ago

alanwaketan commented on 2023-12-12

[SPMD] Add a test case for mark_shard scalar tensors

8e935bcf

Support scalar tensors

d5b9662a

nit

22279011

alanwaketan force pushed from 4d6a508c to 22279011 2 years ago

alanwaketan added backport_2.2

jonb377 approved these changes on 2023-12-12

alanwaketan merged 45a9e0e9 into master 2 years ago

yeounoh commented on 2023-12-12

Reviewers

jonb377

yeounoh

JackCaoG

Assignees

alanwaketan

Labels

backport_2.2

Milestone

No milestone