diff options
| -rw-r--r-- | boltzgen/utility/optimizations.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/boltzgen/utility/optimizations.py b/boltzgen/utility/optimizations.py index 6dc23e9..0acf3e8 100644 --- a/boltzgen/utility/optimizations.py +++ b/boltzgen/utility/optimizations.py @@ -12,7 +12,7 @@ expand_neg_square = ReplaceOptim( lambda p: UnevaluatedExpr(Mul(p.base, p.base, evaluate = False)) ) -custom = [ +custom = cse_main.basic_optimizations + [ (expand_pos_square, expand_pos_square), (expand_neg_square, expand_neg_square) -] + cse_main.basic_optimizations +] |