From fecc58cc2b9de0f1263e4fc5e9cf30a0a872923a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Szymon=20O=C5=BC=C3=B3g?= <szymon.ozog@gmail.com>
Date: Sat, 19 Aug 2023 13:54:21 +0200
Subject: [PATCH] proper function name

---
 tinygrad/runtime/ops_triton.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tinygrad/runtime/ops_triton.py b/tinygrad/runtime/ops_triton.py
index 2f53bb5428..26390cca8d 100644
--- a/tinygrad/runtime/ops_triton.py
+++ b/tinygrad/runtime/ops_triton.py
@@ -101,7 +101,7 @@ def uops_to_triton(function_name:str, uops:List[UOp]):
         else:
           raise NotImplementedError(f"unimplemented: {uop}")
 
-    prg = "@triton.jit\ndef fxn("+','.join(f"data{i}" for i in range(len(bufs)))+"):\n"
+    prg = f"@triton.jit\ndef {function_name}("+','.join(f"data{i}" for i in range(len(bufs)))+"):\n"
     prg += '\n'.join(kernel)
     return prg, global_size, local_size