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