code: def check_topo(top_module, topo: Topo):
      input_partition = topo.get_input_partition()
      mid_partitions = topo.get_mid_partitions()
  
      check_input(top_module, input_partition)
      for part_id, submod in mid_partitions.items():
          check_submod(top_module, part_id, submod)
  
file_path: ColossalAI/tests/test_fx/test_pipeline/test_topo/topo_utils.py
imports: import random
  import numpy
  import torch
  from torch.fx import GraphModule
  from colossalai.fx import ColoTracer
  from colossalai.fx.passes.adding_split_node_pass import balanced_split_pass, split_with_split_nodes_pass
  from colossalai.legacy.pipeline.middleware import Partition, Topo
  from colossalai.legacy.pipeline.middleware.adaptor import get_fx_topology
label_desc: Use image node_id 4 for calling a global function with example usage: check_topo(top_module, topo) without return types