code: def test_torch_ddp_plugin():
      spawn(run_torch_amp, 1)
  
file_path: ColossalAI/tests/test_booster/test_mixed_precision/test_fp16_torch.py
imports: import torch
  from torch.optim import Adam
  import colossalai
  from colossalai.booster.mixed_precision import FP16TorchMixedPrecision
  from colossalai.testing import rerun_if_address_is_in_use, spawn
  from tests.kit.model_zoo import model_zoo
label_desc: Use image node_id 2 for calling a global function with example usage: test_torch_ddp_plugin() without return types