From cc0041cb8cf39cac8bf423b61ecd4b088cb20480 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Mon, 17 Mar 2025 16:30:29 +0800
Subject: [PATCH] padding

---
 extra/replay_pkl.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extra/replay_pkl.py b/extra/replay_pkl.py
index 752dd70b80..b22a851725 100644
--- a/extra/replay_pkl.py
+++ b/extra/replay_pkl.py
@@ -50,6 +50,6 @@ if __name__ == "__main__":
             k.hand_coded_optimizations()
           #if knum in [5]: k.apply_opt(Opt(OptOps.UPCAST, 1, 2))
         p2 = k.to_program()
-        new_ei = replace(ei, prg=CompiledRunner(p2), bufs=[Buffer("DSP", 128+b.size*2, b.dtype).view(b.size, b.dtype, 128) for b in ei.bufs])
+        new_ei = replace(ei, prg=CompiledRunner(p2), bufs=[Buffer("DSP", 1024+b.size*2, b.dtype).view(b.size, b.dtype, 512) for b in ei.bufs])
         new_ei.run()
       knum += 1