From 52364231dc240e022cbc12d41586b004b52be99a Mon Sep 17 00:00:00 2001 From: George Hotz Date: Thu, 27 Mar 2025 17:12:10 +0800 Subject: [PATCH] fast kernel 1 --- extra/replay_pkl.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/extra/replay_pkl.py b/extra/replay_pkl.py index d63154fb5d..b12cd889d1 100644 --- a/extra/replay_pkl.py +++ b/extra/replay_pkl.py @@ -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: