sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions 8 Activity

no copy

Browse Source
This commit is contained in:
geohot
2025-08-07 13:13:50 -07:00
parent aaec715fa8
commit 174747efb3
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/schedule/kernelize.py
+2 -1
View File
@@ -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([