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

lowerer is kernel [run_process_replay] (#5437)

Browse Source
This commit is contained in:
George Hotz
2024-07-12 18:50:55 -07:00
committed by GitHub
parent b8342fb085
commit 03c2dc8bd7
33 changed files with 215 additions and 213 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.lowerer import Lowerer
from tinygrad.codegen.kernel import Kernel
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 = Lowerer(ast)
lin = Kernel(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 = Lowerer(test_lin.ast)
unoptimized_lin = Kernel(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)