code: def auto_activation_checkpoint_batchsize_benchmark():
      spawn(_benchmark, 1)
  
file_path: colossalai/examples/tutorial/auto_parallel/auto_ckpt_batchsize_test.py
imports: from copy import deepcopy
  from functools import partial
  import torch
  import torchvision.models
  from bench_utils import bench, data_gen_resnet
  import colossalai
  from colossalai.auto_parallel.checkpoint import CheckpointSolverRotor
  from colossalai.fx import metainfo_trace, symbolic_trace
  from colossalai.testing import spawn
label_desc: Use image node_id 2 for calling a global function with example usage: auto_activation_checkpoint_batchsize_benchmark() without return types