mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

fast kernel 1

Browse source
This commit is contained in:
George Hotz 2025-03-27 17:12:10 +08:00
commit 52364231dc
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
extra/replay_pkl.py
View file

@ -28,6 +28,8 @@ if __name__ == "__main__":
k = beam_search(k, dsp_bufs, BEAM.value, bool(getenv("BEAM_ESTIMATE", 1)))
elif not getenv("NOOPT"):
if knum == 1:
k.apply_opt(Opt(OptOps.UNROLL, 2, 0))
# more UNROLLs aren't working well here, but they should be
k.apply_opt(Opt(OptOps.UPCAST, 2, 32))
k.apply_opt(Opt(OptOps.UPCAST, 1, 4))
elif knum == 66: