xla
Make partition_spec as tuple to fix issue #8522
#8613

Merged

Make partition_spec as tuple to fix issue #8522 #8613

tengyifei merged 1 commit into pytorch:master from vealocia:fix_spmd_with_scalar

make partition_spec as tuple in case any scalar exists in model's par…

0995ec0f

tengyifei requested a review from

tengyifei 284 days ago

tengyifei approved these changes on 2025-01-25

zpcore added tpuci

tengyifei removed tpuci

tengyifei merged 79e4e72f into master 284 days ago

Reviewers

tengyifei

Assignees

No one assigned

Labels

None yet

Milestone

No milestone