xla
Implement XLAShardedTensor.redistribute and test
#9529

Merged

Implement XLAShardedTensor.redistribute and test #9529

qihqi merged 2 commits into pytorch:master from aws-cph:aws-cph_dtensor_redistribute

Implement XLAShardedTensor.redistribute and test

8cfa637b

qihqi approved these changes on 2025-08-01

fhaolinaws commented on 2025-08-01

Parameterize tests and test for Partial

75bf4027

fhaolinaws approved these changes on 2025-08-04

qihqi enabled auto-merge (squash) 93 days ago

qihqi merged 43589c03 into master 93 days ago

aws-cph deleted the aws-cph_dtensor_redistribute branch 91 days ago

Reviewers

fhaolinaws

qihqi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone