From d41836f1353fa1d56703c90b2de84b146574a3ec Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Thu, 5 Feb 2026 21:55:44 -0500
Subject: [PATCH] remove KERNEL special case in realize_assign [pr] (#14573)

---
 tinygrad/schedule/indexing.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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