From 4e18f4e7aee69d9fe570c49c5b519d92d90f181f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?= Date: Tue, 22 Aug 2023 08:32:22 +0200 Subject: [PATCH] remove print and starting whitespace --- tinygrad/renderer/triton.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tinygrad/renderer/triton.py b/tinygrad/renderer/triton.py index d9dd763434..9f498045f0 100644 --- a/tinygrad/renderer/triton.py +++ b/tinygrad/renderer/triton.py @@ -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)