mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

map BUFFER to Metadata [pr] (#7324)

Browse source
This commit is contained in:
qazal 2024-10-27 20:31:04 +02:00 • committed by GitHub
commit adcdaa17bb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tinygrad/engine/schedule.py
View file

@ -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