From 42dfd836cef70a8e39d5175a34ac87e12930e7af Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 4 Aug 2026 11:20:19 -0700 Subject: [PATCH] contig nonsense --- tinygrad/schedule/rangeify.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tinygrad/schedule/rangeify.py b/tinygrad/schedule/rangeify.py index 38066d865e..887a75b0f1 100644 --- a/tinygrad/schedule/rangeify.py +++ b/tinygrad/schedule/rangeify.py @@ -6,11 +6,11 @@ from tinygrad.uop.ops import PatternMatcher, UPat, Ops, UOp, resolve, GroupOp, K from tinygrad.uop.ops import graph_rewrite, sint, AxisType, BottomUpGate, profile_matches, identity_element from tinygrad.uop.symbolic import symbolic from tinygrad.uop.movement import mop_cleanup -from tinygrad.helpers import prod, getenv, dedup, all_int, DEBUG, SPLIT_REDUCEOP, DEBUG_RANGEIFY, VIZ, MAX_KERNEL_BUFFERS +from tinygrad.helpers import prod, getenv, dedup, all_int, DEBUG, SPLIT_REDUCEOP, VIZ, MAX_KERNEL_BUFFERS from tinygrad.helpers import PCONTIG, FLOAT16, OPENPILOT_HACKS, argsort, partition, get_single_element from tinygrad.codegen.simplify import pm_flatten_range, pm_reduce_simplify from tinygrad.codegen.opt import Opt -from tinygrad.schedule.indexing import run_rangeify, BufferizeOpts, IndexingContext, apply_movement_op +from tinygrad.schedule.indexing import BufferizeOpts, IndexingContext, apply_movement_op from tinygrad.schedule.multi import multi_pm from tinygrad.schedule.allreduce import create_allreduce_function @@ -649,7 +649,7 @@ def get_kernel_graph(sink:UOp) -> UOp: subs: dict[UOp, UOp] = {} for u in tsink.toposort(): u2 = u.replace(src=tuple(subs.get(s, s) for s in u.src)) - subs[u] = u2.alu(Ops.CONTIGUOUS, u2) if u in contig else u2 + subs[u] = u2.alu(Ops.CONTIGUOUS) if u in contig else u2 tsink = subs[tsink] tsink = graph_rewrite(tsink, pm_copy_to_store, ctx=itertools.count(0), bottom_up=True, name="convert copy to store")