From adcdaa17bbb99bbb9fabfa69e8aa611f7a075bfc Mon Sep 17 00:00:00 2001
From: qazal <77887910+Qazalin@users.noreply.github.com>
Date: Sun, 27 Oct 2024 20:31:04 +0200
Subject: [PATCH] map BUFFER to Metadata [pr] (#7324)

---
 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 65e169ad42..5c67f78576 100644
--- a/tinygrad/engine/schedule.py
+++ b/tinygrad/engine/schedule.py
@@ -73,7 +73,7 @@ def to_uop(buf:LazyBuffer, outputs:List[LazyBuffer], ctx:ScheduleContext, metada
   elif buf.op is UnaryOps.BITCAST: ret = UOp(UOps.BITCAST, dtype, src)
   else: ret = UOp(UOps.ALU, dtype, src, buf.op)
   cache[buf] = ret
-  if buf.metadata is not None: metadata[ret] = buf.metadata
+  if buf.metadata is not None: metadata[ubuf] = buf.metadata
   return ret
 
 # **** AST graph rewrite