code: def get_higher_dtype(a, b):
      a = upcast_if_fp8(a)
      b = upcast_if_fp8(b)
      if a is b:
          return a
  
      assert a in _ordered_datatypes
      assert b in _ordered_datatypes
  
      for d in _ordered_datatypes:
          if a is d:
              return b
          if b is d:
              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 2 for calling a global function with example usage: get_higher_dtype(a, b) and returns: a, b, a