code: def linear_func(
      input,
      weight,
      bias,
      add_bias,
      do_flash_attn,
      num_heads,
      transposed_mode=False,
  ):
      assert not transposed_mode and not do_flash_attn
      qkv_out = matmul_ext.matmul(
          input,
          weight,
          bias=(bias if add_bias else None),
          activation="",
          use_triton=True,
      )
  
      return qkv_out
  
file_path: DeepSpeed/deepspeed/ops/transformer/inference/triton/ops.py
imports: import deepspeed
  from deepspeed.ops.op_builder import InferenceBuilder
  import deepspeed.ops.transformer.inference.triton.matmul_ext
  from deepspeed.ops.transformer.inference.triton.layer_norm import layer_norm, layer_norm_residual
label_desc: Use image node_id 3 for calling a global function with example usage: linear_func(input, weight, bias, add_bias, do_flash_attn, num_heads, transposed_mode) and returns: qkv_out