class_docstr: Template kernels defined by CUDA / Cutlass in C++.
class_docstr_tok: ["Template","kernels","defined","by","CUDA","\/","Cutlass","in","C++","."]
docstr: Returns arg name of a given input or output node.
docstr_tok: ["Returns","arg","name","of","a","given","input","or","output","node","."]
code: def arg_name(self, node: IRNode) -> Optional[str]:
      """
      Returns arg name of a given input or output node.
      """
      if node is None:
          return None
      return {
          **self.args.input_buffers,
          **self.args.output_buffers,
      }.get(node.get_name(), None)
  
imports: import logging
  from typing import Callable, Dict, List, Optional, TYPE_CHECKING
  from ...None import ir
  from ...autotune_process import CUDABenchmarkRequest
  from ...ir import Buffer, CUDATemplateBuffer, IRNode, Layout, TensorBox
  from ...select_algorithm import ChoiceCaller
  from ...utils import sympy_product
  from ...virtualized import V
  from ..common import IndentedBuffer, Kernel, OpOverrides
  from ..cpp import CppPrinter, DTYPE_TO_CPP
label_desc: Use image node_id 2 for calling the CUDATemplateKernel obj's underlying member method code with example usage: obj.arg_name(node) and returns: dict, None