mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

no pcontig

Browse source
This commit is contained in:
George Hotz 2025-08-15 22:05:17 -07:00
commit 06fe3a2d57
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tinygrad/schedule/rangeify.py
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