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

test: add fuzz_matmul and better debugging for simple_matmul (#4199)

Browse Source 
also show unoptimized shape in verify_kernel
This commit is contained in:
Francis Lam
2024-04-16 20:40:31 -07:00
committed by GitHub
parent ba8602612b
commit c91b7b1739
3 changed files with 61 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/external/verify_kernel.py vendored
View File

@@ -3,6 +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.features.graph import print_tree
from tinygrad.features.search import time_linearizer
@@ -43,7 +44,9 @@ if __name__ == "__main__":
print_tree(op)
print(op)
print(test_lin.applied_opts)
print(test_lin.colored_shape())
unoptimized_lin = Linearizer(*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)
if msg != "PASS":
failed_ids.append(i)