nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

no pcontig

Browse Source
This commit is contained in:
geohot
2025-08-15 22:05:17 -07:00
parent 778358eff6
commit 06fe3a2d57
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/schedule/rangeify.py
+7 -3
View File
@@ -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