code: def test_split_tensor_into_1d_equal_chunks():
      Utils.initialize_model_parallel(
          tensor_model_parallel_size=2, pipeline_model_parallel_size=4
      )
      input_tensor = torch.rand((3, 4))
      output_tensor = util.split_tensor_into_1d_equal_chunks(
          input_tensor
      )
      if rank % 2 == 0:
          start = 0
          end = int(input_tensor.numel() / 2)
      else:
          start = int(input_tensor.numel() / 2)
          end = input_tensor.numel()
  
      assert torch.equal(
          output_tensor, input_tensor.flatten()[start:end]
      )
      Utils.destroy_model_parallel()
  
file_path: megatron-lm/tests/unit_tests/tensor_parallel/test_tensor_parallel_utils.py
imports: import torch
  import megatron.core.tensor_parallel.utils
  import megatron.core.parallel_state
  from tests.unit_tests.test_utilities import Utils
label_desc: Use image node_id 2 for calling a global function with example usage: test_split_tensor_into_1d_equal_chunks() without return types