From efdf08f3e29b19ee7c547c8ff3755eeb5a7d386f Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Thu, 7 Aug 2025 15:12:00 -0700
Subject: [PATCH] global rangeify

---
 tinygrad/schedule/rangeify.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tinygrad/schedule/rangeify.py b/tinygrad/schedule/rangeify.py
index 5c60070829..690886e7c2 100644
--- a/tinygrad/schedule/rangeify.py
+++ b/tinygrad/schedule/rangeify.py
@@ -82,4 +82,7 @@ pm_rangeify = PatternMatcher([
   # move MAP through elementwise ALU
   (UPat(Ops.INDEX, src=(UPat(GroupOp.Elementwise.union({Ops.STORE})),), allow_any_len=True, name="x"),
    lambda x: x.src[0].replace(src=tuple([UOp(Ops.INDEX, dtype=s.dtype, src=(s,)+x.src[1:]) for s in x.src[0].src]))),
+
+  # CONST can't have axes
+  (UPat(Ops.INDEX, src=(UPat(Ops.CONST,name="c"),)), lambda c: c),
 ])
\ No newline at end of file