Merge pull request #12 from daisyden/xiangdong/fix

Update test_dtensor.py

Author: zxd1997066

test/distributed/tensor/test_dtensor.py

@@ -942,7 +942,7 @@ def test_split_tensor_1D(self) -> None:
         mesh = self.build_device_mesh()
         shard_placement = Shard(0)
 
-        for size in range(8):
+        for size in range(self.world_size):
             tensor = self._create_tensor(size)
             splitted_tensor_list, pad_sizes = shard_placement._split_tensor(
                 tensor,