From 463cd763efad067bed2dc41b24b88b7b35cc4db9 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 4 Aug 2026 16:58:14 -0700 Subject: [PATCH] no reduce ranges (still works for elu) --- tinygrad/schedule/rangeify.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tinygrad/schedule/rangeify.py b/tinygrad/schedule/rangeify.py index cc9617a60b..9606ffb657 100644 --- a/tinygrad/schedule/rangeify.py +++ b/tinygrad/schedule/rangeify.py @@ -681,7 +681,8 @@ def get_kernel_graph(sink:UOp) -> UOp: indexes.setdefault(u.src[0], []).append(u) subs = {} for k,v in indexes.items(): - if len(v) == 1 and k.arg.coeff <= 4: + # don't move REDUCE ranges up (real?) + if len(v) == 1 and all(all([r.arg[-1] == AxisType.WEAK for r in s.ranges]) for s in v[0].src[1:]): for old_r, new_r in zip(k.src[1:], v[0].src[1:]): subs[old_r] = new_r if not len(subs): break