github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

s/Linearizer/Lowerer [run_process_replay] (#5428)

Browse Source
This commit is contained in:
George Hotz
2024-07-12 15:54:07 -07:00
committed by GitHub
parent 6707c778d0
commit 870dc8c350
32 changed files with 159 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/external/verify_kernel.py vendored
View File

@@ -3,7 +3,7 @@ from collections import defaultdict
from extra.optimization.helpers import kern_str_to_lin
from test.external.fuzz_linearizer import compare_linearizer
from tinygrad.helpers import colored
from tinygrad.codegen.linearizer import Linearizer
from tinygrad.codegen.lowerer import Lowerer
from tinygrad.engine.graph import print_tree
from tinygrad.engine.search import time_linearizer
@@ -37,7 +37,7 @@ if __name__ == "__main__":
import pickle
with open(args.pkl, 'rb') as file:
(ast, applied_opts,) = pickle.load(file)
lin = Linearizer(ast)
lin = Lowerer(ast)
for opt in applied_opts:
lin.apply_opt(opt)
test_lins = [lin]
@@ -55,7 +55,7 @@ if __name__ == "__main__":
print_tree(op)
print(op)
print(test_lin.applied_opts)
unoptimized_lin = Linearizer(test_lin.ast)
unoptimized_lin = Lowerer(test_lin.ast)
unoptimized_lin.required_optimizations()
print(f"{unoptimized_lin.colored_shape()} -> {test_lin.colored_shape()}")
(msg,rb,vv,gt) = compare_linearizer(test_lin, None, None, None, rtol=args.rtol, atol=args.atol)