code: def colo_set_cpu_memory_capacity(size: int) -> None:
      global _GLOBAL_CPU_MEM_CAPACITY
      mem_info = _get_cpu_memory_info()
      total_size = mem_info.total
      if size <= total_size:
          _GLOBAL_CPU_MEM_CAPACITY = size
      else:
          _GLOBAL_CPU_MEM_CAPACITY = total_size
  
file_path: colossalai/colossalai/legacy/utils/memory.py
imports: import gc
  from collections import namedtuple
  import psutil
  import torch
  import torch.distributed
  from packaging import version
  from colossalai.legacy.core import global_context
  from colossalai.logging import get_dist_logger
  from colossalai.utils import get_current_device
label_desc: Use image node_id 7 for calling a global function with example usage: colo_set_cpu_memory_capacity(size) without return types