github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

dedup

Browse Source
This commit is contained in:
George Hotz
2025-09-30 16:18:40 +08:00
parent b69b428e9f
commit e090a2ad47
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/schedule/rangeify.py
View File

@@ -402,7 +402,7 @@ def remove_bufferize(src:UOp, buf:UOp, idx:UOp):
ran = src.toposort(gate=red_gate)
# if this is generated from multiple buffers, don't remove this buffer
if len(accessed_buffers) > 1: return None
if len(dedup([x.src[0] for x in accessed_buffers])) > 1: return None
# const reduce is okay
def okay_reduce(x:UOp): return all(y.op not in {Ops.BUFFER, Ops.COPY} for y in x.sparents)