From 996d0ac1d2a41385dde43eb5539a0d0cfe2bfb2d Mon Sep 17 00:00:00 2001 From: George Hotz Date: Mon, 31 Mar 2025 14:17:19 +0800 Subject: [PATCH] multicore all the way --- tinygrad/codegen/kernel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tinygrad/codegen/kernel.py b/tinygrad/codegen/kernel.py index 6ba081a0a2..fc85f6f40a 100644 --- a/tinygrad/codegen/kernel.py +++ b/tinygrad/codegen/kernel.py @@ -482,9 +482,9 @@ class Kernel: k.apply_opt(Opt(OptOps.UPCAST, 0, 128)) # make all non first dimensions local - if getenv("MULTICORE", 1): + if getenv("MULTICORE", 0) and len(k.full_shape) >= 1 and k.full_shape[0] > 1: if k.full_shape[0]%2 == 1: k.apply_opt(Opt(OptOps.PADTO, 0, 2)) - k.apply_opt(Opt(OptOps.LOCAL, 0, k.full_shape[0]//2)) + if k.full_shape[0] > 2: k.apply_opt(Opt(OptOps.LOCAL, 0, k.full_shape[0]//2)) for i in range(1, k.first_reduce-1): k.apply_opt(Opt(OptOps.LOCAL, 1, 0)) else: for i in range(1, k.first_reduce): k.apply_opt(Opt(OptOps.LOCAL, 1, 0))