code: def test_cache_manager_dist():
      spawn(
          create_cache_manager,
          4,
          batch_size=BATCH_SIZE,
          input_len=INPUT_LEN,
          output_len=OUTPUT_LEN,
          layer_num=LAYER_NUM,
          head_num=HEAD_NUM,
          head_dim=HEAD_DIM,
      )
  
file_path: colossalai/tests/test_infer/test_kvcache_manager.py
imports: import os
  import pytest
  import torch
  from packaging import version
  from colossalai.inference.kv_cache import MemoryManager
  from colossalai.logging import disable_existing_loggers
  from colossalai.testing import rerun_if_address_is_in_use, spawn
label_desc: Use image node_id 2 for calling a global function with example usage: test_cache_manager_dist() without return types