From 174747efb33f62bae0245fabe2e67780f703ed4a Mon Sep 17 00:00:00 2001 From: George Hotz Date: Thu, 7 Aug 2025 13:13:50 -0700 Subject: [PATCH] no copy --- tinygrad/schedule/kernelize.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tinygrad/schedule/kernelize.py b/tinygrad/schedule/kernelize.py index f567420fe7..95beb172be 100644 --- a/tinygrad/schedule/kernelize.py +++ b/tinygrad/schedule/kernelize.py @@ -240,7 +240,8 @@ new_fusion = PatternMatcher([ (UPat(Ops.FUSE, src=(UPat(name="s"),)), lambda s: s.replace(src=tuple([y.fuse() for y in s.src]))), # remove CONTIGUOUS if there's no BUFFER upsteam (UPat(Ops.CONTIGUOUS, name="c"), - lambda c: None if c.tag != 2 or any(x.op in GroupOp.UnsafePad.union({Ops.BUFFER}) for x in c.toposort()) else c.src[0]), + lambda c: None if c.tag != 2 or c.src[0].op is Ops.COPY or + any(x.op in GroupOp.UnsafePad.union({Ops.BUFFER}) for x in c.toposort()) else c.src[0]), ]) finalize_contiguous = PatternMatcher([