code: def upcast_if_fp8(a):
      if "fp8" in str(a):
          return torch.float16
      return a
  
file_path: triton/python/triton/ops/matmul.py
imports: import torch
  from ..None import Config, autotune, cdiv, heuristics, jit
  from ..None import language
  from .matmul_perf_model import early_config_prune, estimate_matmul_time
label_desc: Use image node_id 1 for calling a global function with example usage: upcast_if_fp8(a) and returns: a, torch