From 09c2c64ca077e7ff6c871dabc2826ec315f74186 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Wed, 18 Feb 2026 10:32:10 +0800
Subject: [PATCH] just buffer

---
 tinygrad/engine/schedule.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py
index c87785a99c..51f9e25043 100644
--- a/tinygrad/engine/schedule.py
+++ b/tinygrad/engine/schedule.py
@@ -148,7 +148,7 @@ pm_post_sched_cache = PatternMatcher([
 # rewrite all contiguous to assign
 def contig_to_assign(ctx:dict[UOp,UOp|None], x:UOp):
   # existing ASSIGN already has a target buffer, add it to the map
-  if x.op is Ops.ASSIGN:
+  if x.op is Ops.ASSIGN and x.src[0].base.op is Ops.BUFFER:
     ctx[x] = x.src[0]
     return None
   # for contiguous or in buffer_map explicitly