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={}
          )
      view_node = list(x_node.users.keys())[0]
      new_args = list(view_node.args)
      new_args[0] = shard_node
      view_node.args = tuple(new_args)
      return gm
  
file_path: ColossalAI/tests/test_auto_parallel/test_pass/test_node_converting_pass.py
imports: import torch
  from colossalai.auto_parallel.passes.runtime_preparation_pass import node_args_converting_pass
  from colossalai.device.device_mesh import DeviceMesh
  from colossalai.fx.graph_module import ColoGraphModule
  from colossalai.fx.tracer import ColoTracer
  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