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

remove print and starting whitespace

Browse Source
This commit is contained in:
Szymon Ożóg
2023-08-22 08:32:22 +02:00
parent 1d5cd82ed5
commit 4e18f4e7ae
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/renderer/triton.py
View File

@@ -1,4 +1,3 @@
from typing import Dict, List, Final, Callable
from tinygrad.ops import UnaryOps, BinaryOps, TernaryOps, Op
from tinygrad.helpers import dtypes, ImageDType
@@ -74,9 +73,6 @@ def uops_to_triton(function_name:str, uops:List[UOp]):
prg = f"@triton.jit\ndef {function_name}("+','.join(f"data{i}" for i in range(len(bufs)))+"):\n"
prg += '\n'.join(kernel)
print(prg)
acc_local_size = 1
for x in local_size: acc_local_size *= next_power_of_2(x)
local_size = [acc_local_size] + [1] * (len(local_size) - 1)