From 91cae8966f5ac5c388fe0f9bcaacf924dd38ed6c Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Tue, 14 Jan 2020 10:31:32 +0100
Subject: [PATCH] Added option to pass options to sympy CSE through
simplification rule
---
pystencils/simp/simplifications.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pystencils/simp/simplifications.py b/pystencils/simp/simplifications.py
index e2fbbbf80..5d9b819d5 100644
--- a/pystencils/simp/simplifications.py
+++ b/pystencils/simp/simplifications.py
@@ -29,7 +29,7 @@ def sort_assignments_topologically(assignments: Sequence[Union[Assignment, Node]
return [assignments[i] for i in sp.topological_sort((range(len(assignments)), edges))]
-def sympy_cse(ac):
+def sympy_cse(ac, **kwargs):
"""Searches for common subexpressions inside the assignment collection.
Searches is done in both the existing subexpressions as well as the assignments themselves.
@@ -40,7 +40,7 @@ def sympy_cse(ac):
all_assignments = [e for e in chain(ac.subexpressions, ac.main_assignments) if isinstance(e, Assignment)]
other_objects = [e for e in chain(ac.subexpressions, ac.main_assignments) if not isinstance(e, Assignment)]
- replacements, new_eq = sp.cse(all_assignments, symbols=symbol_gen)
+ replacements, new_eq = sp.cse(all_assignments, symbols=symbol_gen, **kwargs)
replacement_eqs = [Assignment(*r) for r in replacements]
--
GitLab