<!-- ROCM-SKIP:test/distributed/_shard/sharded_tensor/test_sharded_tensor.py::TestShardedTensorChunked::test_multiple_local_shards --> ## Context - Source file: `test/distributed/_shard/sharded_tensor/test_sharded_tensor.py` - Test class: `TestShardedTensorChunked` - Test name: `test_multiple_local_shards` - Defined at line: 947 - Parent issue: #780 - Code reference: https://github.com/chinmaydk99/pytorch/blob/main/test/distributed/_shard/sharded_tensor/test_sharded_tensor.py