From 4d40de867b9becefd9526621bbd49d8d09f42ba3 Mon Sep 17 00:00:00 2001 From: chenyu Date: Thu, 22 Aug 2024 16:45:49 -0400 Subject: [PATCH] remove redundant `c1-(x+c2)` rule [run_process_replay] (#6243) --- tinygrad/codegen/uopgraph.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tinygrad/codegen/uopgraph.py b/tinygrad/codegen/uopgraph.py index cb068d8ed9..3869ec75a0 100644 --- a/tinygrad/codegen/uopgraph.py +++ b/tinygrad/codegen/uopgraph.py @@ -251,8 +251,6 @@ constant_folder = PatternMatcher([ (NOp(UOps.REDUCE, src=(NOp.var('idx').eq(NOp(UOps.RANGE, name="rng")).where( NOp(UOps.LOAD, src=(NOp.var("buf"), NOp.var('add')+NOp.var('mul')*NOp(UOps.RANGE, name="rng")), name="ld"), NOp.const(None, 0.0)),), arg=BinaryOps.ADD, name="reduce", allow_any_len=True), index_collapse), - # other arange folders - (NOp.cvar("c1") - (NOp.var("x") + NOp.cvar("c2")), lambda c1, c2, x: (c1-c2)-x), # c1 - (x + c2) -> (c1-c2) - x # max folding (NOp.max(NOp.var('x'), NOp.var('y')), lambda x,y: x if x.vmin.arg >= y.vmax.arg else y if x.vmax.arg <= y.vmin.arg else None), # const rules