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

proper function name

Browse Source
This commit is contained in:
Szymon Ożóg
2023-08-19 13:54:21 +02:00
parent b624a374b9
commit fecc58cc2b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/ops_triton.py
View File

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