tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 11:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

rewrite name

Browse Source
This commit is contained in:
geohot
2025-08-05 20:28:15 -07:00
parent 07f45be2a7
commit 4689d29f49
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/schedule/kernelize.py
+1 -1
View File
@@ -176,7 +176,7 @@ def fix_kernel_ast(k:UOp) -> UOp|None:
# TODO: move these to codegen
ast = graph_rewrite(ast, view_left, name="Main View Left")
ast = graph_rewrite(ast, view_right, name="Main View Right")
ast = graph_rewrite(ast, view_left+fix_kernel_ops, bottom_up=True, name="replace buffer")
ast = graph_rewrite(ast, view_left+fix_kernel_ops, bottom_up=True, name="Finalize Kernel")
return k.replace(arg=Kernel(ast, k.arg.metadata))
create_ast = PatternMatcher([(UPat(Ops.KERNEL, name="k"), fix_kernel_ast),])