code: def pipeline_pass_test_helper(model, data, pass_func):
      origin_output = model(data)
      symbolic_traced = symbolic_trace(model)
      annotated_model = pass_func(symbolic_traced, PIPELINE_SIZE)
      split_model, split_submodules = split_with_split_nodes_pass(
          annotated_model
      )
      output = split_model(data)
      assert output.equal(origin_output)
  
file_path: colossalai/tests/test_fx/test_pipeline_passes.py
imports: import torch
  from torch.fx import symbolic_trace
  from colossalai.fx.passes.adding_split_node_pass import balanced_split_pass, balanced_split_pass_v2, split_with_split_nodes_pass, uniform_split_pass
  from colossalai.testing import clear_cache_before_run
label_desc: Use image node_id 1 for calling a global function with example usage: pipeline_pass_test_helper(model, data, pass_func) without return types