From f3cb4c3eef9a0da3e650a48fc4e6dd09c6b60bf5 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 1 Apr 2025 23:44:44 +0800 Subject: [PATCH] oops --- tinygrad/codegen/kernel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/codegen/kernel.py b/tinygrad/codegen/kernel.py index 8c274d57f0..e9560700c4 100644 --- a/tinygrad/codegen/kernel.py +++ b/tinygrad/codegen/kernel.py @@ -447,7 +447,7 @@ class Kernel: k.apply_opt(Opt(OptOps.UNROLL, 0, 0)) k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-3, 32)) if k.full_shape[len(k.full_shape)-4]%4 == 0: - if len(k.full_shape)-4 <= 8: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 0)) + if k.full_shape[len(k.full_shape)-4] <= 8: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 0)) else: k.apply_opt(Opt(OptOps.UPCAST, len(k.full_shape)-4, 4)) # if this is small, swap it # NOTE: this is breaking something (should be fixed w/o padto)