From 06fe3a2d57ce8a94fec1e62ca9adaaa75be7dbc7 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Fri, 15 Aug 2025 22:05:17 -0700
Subject: [PATCH] no pcontig

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

diff --git a/tinygrad/schedule/rangeify.py b/tinygrad/schedule/rangeify.py
index 9423b5fdb7..7b57e67fc0 100644
--- a/tinygrad/schedule/rangeify.py
+++ b/tinygrad/schedule/rangeify.py
@@ -164,13 +164,17 @@ pm_mops = PatternMatcher([
 ])
 
 def map_contiguous(ctx:RangeifyContext, x:UOp, idx:UOp|None=None):
-  if x.arg is None and idx is not None: return None
-  if x.arg is not None and idx is None: return None
+  # NOTE: partial contig is disabled for now
+  #arg = x.arg
+  arg = None
+  if arg is None and idx is not None: return None
+  if arg is not None and idx is None: return None
+
   ranges = []
   new_ranges = []
   passthrough_idx = []
   for i,s in enumerate(x.shape):
-    if x.arg is not None and i not in x.arg:
+    if arg is not None and i not in arg:
       assert idx is not None, "partial contig requires index"
       ranges.append(idx.src[1+i])
       continue