From a11aa78d6b1cb105e8744e81459164fd075ee092 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?= Date: Fri, 18 Aug 2023 16:10:53 +0200 Subject: [PATCH] Disable float4 support --- tinygrad/runtime/ops_triton.py | 1 + 1 file changed, 1 insertion(+) 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