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

use Estimates class [pr] (#8319)

Browse Source 
* use Estimates class [pr]

* frozen dataclass
This commit is contained in:
George Hotz
2024-12-18 10:19:32 -08:00
committed by GitHub
parent 63f195729d
commit 8f95b578f6
8 changed files with 52 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/handcode_opt.py
View File

@@ -109,7 +109,7 @@ if __name__ == "__main__":
choices = []
for lin, nm in lins:
tm = time_linearizer(lin, rawbufs, allow_test_size=False, cnt=10, disable_cache=True)
ops = (prg:=lin.to_program()).op_estimate
ops = (prg:=lin.to_program()).estimates.ops
gflops = sym_infer(ops, {k:k.min for k in lin.ast.variables()})*1e-9/tm
choices.append((tm, gflops, lin, prg, nm))