github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

current_kernel fix

Browse Source
This commit is contained in:
qazal
2024-09-25 17:41:50 +08:00
parent ee59c07719
commit 138713f4c4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/codegen/kernel.py
View File

@@ -704,6 +704,7 @@ class Kernel:
# **** this is the lowerer ****
def linearize(self) -> Kernel:
if TRACK_MATCH_STATS >= 2: _CURRENT_KERNEL.set(self.name)
modified_ast = self.get_optimized_ast()
if DEBUG >= 3:
@@ -713,7 +714,6 @@ class Kernel:
print(self.applied_opts)
verify_ast(modified_ast)
if TRACK_MATCH_STATS >= 2: _CURRENT_KERNEL.set(self.name)
self.uops:List[UOp] = linearize_uop(full_graph_rewrite(ast_to_uop(modified_ast, self.opts), self.opts))
if TRACK_MATCH_STATS >= 2: _CURRENT_KERNEL.set(None)
if DEBUG >= 5: print_uops(self.uops)