code: def get_kernel_dtype(param_dtype):
      if param_dtype == torch.float16:
          return "__half2"
      elif param_dtype == torch.float32:
          return "float"
      else:
          raise Exception("Unrecognized data type: %s" % param_dtype)
  
imports: import torch
  from ..impls.jit_compiler import JitCompiler
label_desc: Use image node_id 1 for calling a global function with example usage: get_kernel_dtype(param_dtype) and returns: str, str