From a8bd26d9bceedf34e34c4aa730643a4fa74d5189 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Thu, 27 Mar 2025 16:39:15 +0800 Subject: [PATCH] full_shape --- extra/replay_pkl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extra/replay_pkl.py b/extra/replay_pkl.py index 3dde042903..1a6af06de5 100644 --- a/extra/replay_pkl.py +++ b/extra/replay_pkl.py @@ -44,13 +44,13 @@ if __name__ == "__main__": #elif k.full_shap[-4] == 7: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 7)) #elif k.full_shape[-4] == 14: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 2)) elif len(k.full_shape) == 3 and k.full_shape[1] == 32: - #if k.full_shape[0]%4 != 0: k.apply_opt(Opt(OptOps.PADTO, 0, 4)) + if k.full_shape[0]%4 != 0: k.apply_opt(Opt(OptOps.PADTO, 0, 4)) # weight without more k.apply_opt(Opt(OptOps.UNROLL, 0, 8)) k.apply_opt(Opt(OptOps.UPCAST, 1, 32)) if k.full_shape[0]%4 == 0: k.apply_opt(Opt(OptOps.UPCAST, 0, 4)) elif len(k.full_shape) == 4 and k.full_shape[2] == 32: - #if k.full_shape[1]%4 != 0: k.apply_opt(Opt(OptOps.PADTO, 1, 4)) + if k.full_shape[1]%4 != 0: k.apply_opt(Opt(OptOps.PADTO, 1, 4)) # weight with more k.apply_opt(Opt(OptOps.UNROLL, 0, 8)) k.apply_opt(Opt(OptOps.UPCAST, 2, 32))