code: def insert_narrow(gm, x_node):
      graph = gm.graph
      with graph.inserting_after(x_node):
          shard_node = graph.create_node(
              "call_method", "narrow", args=(x_node, 0, 0, 2), kwargs={}
          )
      size_node = list(x_node.users.keys())[0]
      size_node.args = (shard_node,)
      return gm
  
file_path: ColossalAI/tests/test_auto_parallel/test_pass/test_size_value_converting_pass.py
imports: import pytest
  import torch
  from colossalai._analyzer.fx.graph_module import ColoGraphModule
  from colossalai._analyzer.fx.passes import shape_prop_pass
  from colossalai._analyzer.fx.tracer.tracer import ColoTracer
  from colossalai.auto_parallel.passes.runtime_preparation_pass import size_value_converting_pass
  from colossalai.device.device_mesh import DeviceMesh
  from colossalai.tensor.sharding_spec import ShardingSpec
  from colossalai.testing import clear_cache_before_run
label_desc: Use image node_id 1 for calling a global function with example usage: insert_narrow(gm, x_node) and returns: gm