1 files changed, 7 insertions, 1 deletions

diff --git a/pyecsca/misc/cache.py b/pyecsca/misc/cache.py
index 385ff4b..5f388ad 100644
--- a/pyecsca/misc/cache.py
+++ b/pyecsca/misc/cache.py
@@ -1,6 +1,6 @@
 """Cache some things."""
 from functools import lru_cache
-from sympy import sympify as _orig_sympify
+from sympy import sympify as _orig_sympify, simplify as _orig_simplify, count_ops
 from public import public
 
 
@@ -10,3 +10,9 @@ def sympify(
     a, locals=None, convert_xor=True, strict=False, rational=False, evaluate=None
 ):
     return _orig_sympify(a, locals, convert_xor, strict, rational, evaluate)
+
+
+@public
+@lru_cache(maxsize=256, typed=True)
+def simplify(expr, ratio=1.7, measure=count_ops, rational=False, inverse=False, doit=True, **kwargs):
+    return _orig_simplify(expr, ratio=ratio, measure=measure, rational=rational, inverse=inverse, doit=doit, **kwargs)