code: def dist_sum(x):
      temp = torch.tensor([x], device=get_current_device())
      dist.all_reduce(temp)
      return temp.item()
  
file_path: colossalai/tests/test_zero/test_gemini/test_chunkv2.py
imports: import pytest
  import torch
  import torch.distributed
  from torch.distributed.distributed_c10d import _get_default_group
  import colossalai
  from colossalai.tensor import ColoParameter
  from colossalai.testing import parameterize, rerun_if_address_is_in_use, spawn
  from colossalai.utils import get_current_device
  from colossalai.zero.gemini import TensorState
  from colossalai.zero.gemini.chunk import Chunk
label_desc: Use image node_id 1 for calling a global function with example usage: dist_sum(x) and returns: temp