xla
[XLA] Add placements property to XLAShardedTensor for DTensor compatibility
#9509

Merged

[XLA] Add placements property to XLAShardedTensor for DTensor compatibility #9509

rpsilva-aws merged 7 commits into pytorch:master from Hoomaaan:placements

jeffhataws requested a review from

rpsilva-aws 102 days ago

jeffhataws requested a review from

zhanyong-wan 102 days ago

jeffhataws requested a review from

bfolie 102 days ago

Hoomaaan changed the title ~~[XLA] Add placements property to XLAShardedTensor for DTensor compati…~~ [XLA] Add placements property to XLAShardedTensor for DTensor compatibility 102 days ago

[XLA] Add placements property to XLAShardedTensor for DTensor compati…

f895549f

Hoomaaan force pushed from 2c6ff64b to f895549f 102 days ago

Fix indentation for yapf

97cf32c4

run git_fix.py to fix yapf error

b32a736f

bfolie commented on 2025-07-25

Remove check for cache

b5d68371

Hoomaaan requested a review from

bfolie 101 days ago

bfolie commented on 2025-07-28

Remove print statements from the tests

230365ca

Hoomaaan requested a review from

bfolie 98 days ago

bfolie approved these changes on 2025-07-28

rpsilva-aws requested changes on 2025-07-29

Modify placements tests

efac29a5

Use class method instead of defining a new invalid func

849d3dde

rpsilva-aws approved these changes on 2025-07-29

rpsilva-aws merged 95bee8f0 into master 96 days ago

Reviewers

rpsilva-aws

bfolie

zhanyong-wan

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

xla [XLA] Add placements property to XLAShardedTensor for DTensor compatibility #9509 Merged

[XLA] Add placements property to XLAShardedTensor for DTensor compatibility #9509

xla
[XLA] Add placements property to XLAShardedTensor for DTensor compatibility
#9509

Merged