nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

does that pass?

Browse Source
This commit is contained in:
geohot
2025-09-08 11:12:46 -07:00
parent 8b944a32eb
commit 18da4cdcbd
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/codegen/opt/postrange.py
+4 -4
View File
@@ -326,10 +326,10 @@ def apply_opts(ctx:Renderer, ast:UOp):
# NOTE: hand_coded_optimizations doesn't support multiblock opts yet
if all(len(u.src) == 1 for u in ast.parents if u.op is Ops.LOAD):
# TODO: why is the returned k from hand_coded_optimizations different than this?
hc_k = hand_coded_optimizations(k)
for opt in hc_k.applied_opts: k.apply_opt(opt)
assert hc_k.applied_opts == k.applied_opts
assert hc_k.get_optimized_ast(name_override="k") == k.get_optimized_ast(name_override="k")
k = hand_coded_optimizations(k)
#for opt in hc_k.applied_opts: k.apply_opt(opt)
#assert hc_k.applied_opts == k.applied_opts
#assert hc_k.get_optimized_ast(name_override="k") == k.get_optimized_ast(name_override="k")
return k.get_optimized_ast(name_override=ast.arg.name if ast.arg is not None and ast.arg.name != "test" else None)
pm_postrange_opt = PatternMatcher([