diff --git a/tinygrad/schedule/indexing.py b/tinygrad/schedule/indexing.py index fd4fbc65b0..d8d4fcfbc9 100644 --- a/tinygrad/schedule/indexing.py +++ b/tinygrad/schedule/indexing.py @@ -19,8 +19,7 @@ def realize_srcs(ctx:dict[UOp, None], rb:UOp) -> None: def realize_assign(ctx:dict[UOp, None], a:UOp) -> None: if a.src[1].op not in ALWAYS_CONTIGUOUS: ctx[a.src[1]] = None - # if it's a kernel, we don't realize it - if a.src[1].op is not Ops.KERNEL: ctx[a] = None + ctx[a] = None pm_generate_realize_map = PatternMatcher([ # always realize SINK src