code: def trace_triton_kernel_wrapper(proxy_mode, func_overload, node_args):
      with disable_proxy_modes_tracing():
          out = func_overload(**node_args)
  
      proxy_args = pytree.tree_map(
          proxy_mode.tracer.unwrap_proxy, node_args
      )
      out_proxy = proxy_mode.tracer.create_proxy(
          "call_function",
          func_overload,
          (),
          proxy_args,
          name=func_overload.__name__ + "_proxy",
      )
      return track_tensor_tree(
          out, out_proxy, constant=None, tracer=proxy_mode.tracer
      )
  
imports: import ast
  import dataclasses
  import threading
  from typing import Any, Dict
  import torch.utils._pytree
  from torch import Tensor
  from torch._C import DispatchKey
  from torch._ops import HigherOrderOperator
  from torch._prims_common import clone_preserve_strides
  from torch._subclasses.fake_tensor import FakeTensorMode
  from torch.fx.experimental.proxy_tensor import disable_proxy_modes_tracing, ProxyTorchDispatchMode, track_tensor_tree
label_desc: Use image node_id 4 for calling a global function with example usage: trace_triton_kernel_wrapper(proxy_mode, func_overload, node_args) and returns: track_tensor_tree