code: def sync_states():
      current_mode = get_current_mode()
      current_states = torch.cuda.get_rng_state()
      set_seed_states(current_mode, current_states)
  
file_path: ColossalAI/colossalai/legacy/context/random/_helper.py
imports: import functools
  from contextlib import contextmanager
  import torch.cuda
  from torch import Tensor
  from ..parallel_mode import ParallelMode
  from .seed_manager import SeedManager
label_desc: Use image node_id 7 for calling a global function with example usage: sync_states() without return types