Method render_kernel

tinygrad/renderer/cstyle.py:316–318

(self, function_name, kernel, bufs, uops, prefix=None)

  ]) + base_rewrite

  def render_kernel(self, function_name, kernel, bufs, uops, prefix=None) -> str:
    if any(uop.dtype.base == dtypes.half for uop in uops): prefix = (["#pragma OPENCL EXTENSION cl_khr_fp16 : enable"] + (prefix or []))
    return super().render_kernel(function_name, kernel, bufs, uops, prefix)

  def aux(self, uops:list[UOp]):
    arg_dtypes:list[list[tuple[int, DType]]] = []