github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Disable float4 support

Browse Source
This commit is contained in:
Szymon Ożóg
2023-08-18 16:10:53 +02:00
parent 2534e8f5a3
commit a11aa78d6b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tinygrad/runtime/ops_triton.py
View File

@@ -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)