code: def test_torch_amp():
      spawn(run_dist, 1)
  
file_path: colossalai/tests/test_legacy/test_amp/test_torch_fp16.py
imports: import copy
  import pytest
  import torch
  import colossalai
  from colossalai.legacy.amp import convert_to_apex_amp, convert_to_torch_amp
  from colossalai.testing import assert_close_loose, clear_cache_before_run, rerun_if_address_is_in_use, spawn
  from tests.kit.model_zoo import model_zoo
label_desc: Use image node_id 3 for calling a global function with example usage: test_torch_amp() without return types