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

fixing the benchmark not printing in handcode resnet50 opt example (#3850)

Browse Source
This commit is contained in:
Anurag Lamsal
2024-03-21 10:40:31 +05:45
committed by GitHub
parent 85691c8e20
commit 4e0819e40b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/handcode_resnet50_opt.py
View File

@@ -62,7 +62,7 @@ if __name__ == "__main__":
choices = []
for lin in lins:
tm = time_linearizer(lin, rawbufs, allow_test_size=False, cnt=10)
gflops = sym_infer(lin.info.flops, {k:k.min for k in lin.ast.vars()})*1e-9/tm
gflops = sym_infer(lin.info.flops, {k:k.min for k in lin.ast[0].vars()})*1e-9/tm
choices.append((tm, gflops, lin.linearize()))
# print all kernels