From 4123920bcc4fda2086fa6de37aed46a125ad1925 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?= Date: Sat, 19 Aug 2023 13:56:37 +0200 Subject: [PATCH] remove deprecated variables --- tinygrad/runtime/ops_triton.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/tinygrad/runtime/ops_triton.py b/tinygrad/runtime/ops_triton.py index 26390cca8d..7e1f57ae46 100644 --- a/tinygrad/runtime/ops_triton.py +++ b/tinygrad/runtime/ops_triton.py @@ -43,9 +43,6 @@ def uops_to_triton(function_name:str, uops:List[UOp]): bufs = [] def kk(s): kernel.append(" "*depth+s) - full_local_shape: Tuple[Any, ...] = () - acc_local_shape = 1 - gid = [f"tl.program_id({i})" for i in range(3)] code_for_op: Final[Dict[Op, Callable]] = { UnaryOps.EXP2: lambda x: f"tl.math.exp2({x})", @@ -69,10 +66,8 @@ def uops_to_triton(function_name:str, uops:List[UOp]): global_size.append(var.max+1) kk(f"{var.expr} = {gid[i]} # {var.max+1}") elif args[1] == "local": - full_local_shape = tuple([var.max+1 for var in args[0]]) assert var.min == 0, "local loop must start at 0" kk(f"{var.expr} = tl.arange({0}, {next_power_of_2(var.max+1)})[{', '.join([':' if i == j else 'None' for j in range(len(args[0]))])}]") - acc_local_shape *= var.max+1 local_size.append(var.max+1) else: kk(f"for {var.expr} in range({var.min}, {var.max+1}):")