code: def test_create_chunk_sharded_tensor(self):
      for size in ((1,), (1, 6), (12,), (12, 6), (25,), (25, 6)):
          tensor = self._create_tensor(*size)
  
          sharded_tensor = _create_chunk_sharded_tensor(
              tensor,
              self.rank,
              self.world_size,
              torch.cuda.device_count(),
              _get_default_group(),
          )
          output = torch.empty(*size).cuda() if self.rank == 0 else None
          sharded_tensor.gather(0, output)
          if self.rank == 0:
              self.assertEqual(tensor, output)
  
file_path: pytorch/test/distributed/fsdp/test_shard_utils.py
imports: import torch
  from torch.distributed.distributed_c10d import _get_default_group
  from torch.distributed.fsdp._shard_utils import _create_chunk_dtensor, _create_chunk_sharded_tensor
  from torch.testing._internal.common_fsdp import FSDPTest
  from torch.testing._internal.common_utils import run_tests
  from torch.testing._internal.distributed._tensor.common_dtensor import DTensorTestBase, skip_if_lt_x_gpu, with_comms
label_desc: Use image node_id 3 for calling the TestShardUtilsDistributed obj's underlying member method code with example usage: obj.test_create_chunk_sharded_tensor() without return types