code: def triton_config(num_stages, num_warps, **kwargs):
      from triton import Config
  
      return Config(kwargs, num_stages=num_stages, num_warps=num_warps)
  
file_path: pytorch/torch/_inductor/kernel/mm_common.py
imports: import functools
  import logging
  from typing import cast, List, Tuple
  import sympy
  import torch
  from torch._inductor.select_algorithm import realize_inputs
  from torch._inductor.virtualized import V
  from ..None import config
  from ..utils import ceildiv, next_power_of_2
label_desc: Use image node_id 1 for calling a global function with example usage: triton_config(num_stages, num_warps) and returns: Config