github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-23 13:58:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

single arange fusion with graph rewrite (#6160)

Browse Source
This commit is contained in:
qazal
2024-08-26 18:18:16 +08:00
committed by GitHub
parent b4381e9777
commit 067aeaeb2f
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

@@ -188,7 +188,7 @@ def push_reduceop_shape(root:UOp) -> Optional[UOp]:
reduceop_fusor = PatternMatcher([
(UPat(UOps.REDUCE_AXIS, src=(UPat(name="rsrc"), UPat(UOps.SWIZZLE, src=(UPat(name="swizzle"),))), name="root"), apply_swizzle),
(UPat(UOps.STORE, name="root"), push_reduceop_shape),
(UPat({UOps.ALU, UOps.CAST, UOps.BITCAST, UOps.STORE}, name="root"), push_reduceop_shape),
])
def _lower_lazybuffer(outs:List[LazyBuffer], realizes:Dict[LazyBuffer, None]) -> LBScheduleItem: