diff --git a/tinygrad/runtime/ops_triton.py b/tinygrad/runtime/ops_triton.py index 30d41f18be..b6e1c8716b 100644 --- a/tinygrad/runtime/ops_triton.py +++ b/tinygrad/runtime/ops_triton.py @@ -100,3 +100,4 @@ def uops_to_triton(function_name:str, uops:List[UOp]): prg += '\n'.join(kernel) return prg, global_size, local_size +TritonBuffer = Compiled(RawCUDABuffer, LinearizerOptions(supports_float4=False ,supports_float4_alu=False), uops_to_triton, TritonProgram) \ No newline at end of file