Skip to content

Commit 55f806b

Browse files
authored
tighter late_buffer_view match [pr] (tinygrad#14456)
src must be len 2 at that point
1 parent d69bc5a commit 55f806b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tinygrad/schedule/rangeify.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -280,10 +280,10 @@ def late_buffer_view(t:UOp, b:UOp):
280280
if len(shape) == 0: offset = x.src[1].arg
281281
else: offset = max(sum(idx.vmin for idx in x.src[1:]), 0)
282282

283-
return b.replace(src=(UOp(Ops.BUFFER_VIEW, t.dtype, (x.base,), (size, offset), tag=t.tag),) + b.src[1:])
283+
return b.replace(src=(UOp(Ops.BUFFER_VIEW, t.dtype, (x.base,), (size, offset), tag=t.tag), b.src[1]))
284284

285285
to_bufferview = PatternMatcher([
286-
(UPat((Ops.BITCAST, Ops.CONTIGUOUS), name="t").f(Ops.BUFFERIZE, allow_any_len=True, name="b"), late_buffer_view),
286+
(UPat(Ops.BUFFERIZE, src=(UPat((Ops.BITCAST, Ops.CONTIGUOUS), name="t"), UPat()), name="b"), late_buffer_view),
287287
(UPat((Ops.BITCAST, Ops.CONTIGUOUS)).f(Ops.BUFFER_VIEW, name="b"), lambda b: b.replace(src=b.src[0].src)),
288288
])
289289

0 commit comments

Comments
 (0)