From de773b593e2bb68f6ae3b8178eab79e9787bdb92 Mon Sep 17 00:00:00 2001 From: chenyu Date: Wed, 14 Aug 2024 13:18:28 -0400 Subject: [PATCH] remove redundant div gcd patterns [run_process_replay] (#6076) covered by generic div_folding --- tinygrad/codegen/uopgraph.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tinygrad/codegen/uopgraph.py b/tinygrad/codegen/uopgraph.py index 8deeb7342e..067fd0af79 100644 --- a/tinygrad/codegen/uopgraph.py +++ b/tinygrad/codegen/uopgraph.py @@ -284,9 +284,6 @@ constant_folder = PatternMatcher([ # # div folding (NOp.var('x') // NOp.cvar('c'), lambda x,c: newx if 0 < c.arg and not dtypes.is_unsigned(x.dtype) and (newx:=div_folding(x,c.arg)) is not None else None), - # mul add div - (((NOp.cvar('c0')*NOp.var('x'))+NOp.var('x2')) // NOp.cvar('c1'), lambda x,x2,c0,c1:\ - x*(c0.arg//g)//(c1.arg//g) if c0.arg > 0 and c1.arg > 0 and (g:=math.gcd(c0.arg,c1.arg)) > 1 and g > x2.vmax.arg and x2.vmin.arg >= 0 else None), # ** mod ** # apply mod to mod input (NOp.var('x') % NOp.cvar('c'), lambda x,c: newx%c if 0 < c.arg and (newx:=mod_folding(x,c.arg)) is not None else None),