code: def implicit_comm_action_apply(gm: torch.fx.GraphModule):
      """
      replace the origin kernel into kernel with implicit communication inside.
      """
  
file_path: colossalai/colossalai/auto_parallel/passes/runtime_preparation_pass.py
imports: import operator
  from typing import Dict, List, Union
  import torch
  from torch.fx.node import Node
  from colossalai._analyzer.fx.node_util import MetaInfo
  from colossalai.auto_parallel.tensor_shard.constants import RESHAPE_FUNC_OP
  from colossalai.auto_parallel.tensor_shard.sharding_strategy import CommType, OperationDataType
  from colossalai.auto_parallel.tensor_shard.solver.strategies_constructor import StrategiesConstructor
  from colossalai.device.device_mesh import DeviceMesh
  from colossalai.tensor.comm_spec import _all_reduce
  from colossalai.tensor.shape_consistency import ShapeConsistencyManager
  from colossalai.tensor.sharding_spec import ShardingSpec
  from .constants import SHAPE_ARGUMENT_OPS
label_desc: Use image node_id 6 for calling a global function with example usage: implicit_comm_action_apply(gm) without return types