Cache sympify results to speedup formula exec.

pyecsca/ec/curve.py

@@ -5,7 +5,8 @@
 from typing import MutableMapping, Union, List, Optional, Dict, Set
 
 from public import public
-from sympy import FF, sympify
+from sympy import FF
+from ..misc.cache import sympify
 
 from .coordinates import CoordinateModel, AffineCoordinateModel
 from .error import raise_unsatisified_assumption

pyecsca/ec/formula.py

@@ -9,7 +9,8 @@
 
 from importlib_resources.abc import Traversable
 from public import public
-from sympy import sympify, FF, symbols, Poly, Rational
+from sympy import FF, symbols, Poly, Rational
+from ..misc.cache import sympify
 
 from .context import ResultAction
 from . import context

pyecsca/ec/params.py

@@ -5,7 +5,7 @@
 """
 import json
 import csv
-from sympy import Poly, FF, symbols, sympify
+from sympy import Poly, FF, symbols
 from astunparse import unparse
 from io import RawIOBase, BufferedIOBase
 from pathlib import Path
@@ -14,6 +14,7 @@
 
 from public import public
 
+from ..misc.cache import sympify
 from .coordinates import AffineCoordinateModel, CoordinateModel
 from .curve import EllipticCurve
 from .error import raise_unsatisified_assumption

pyecsca/misc/cache.py

@@ -0,0 +1,12 @@
+"""Cache some things."""
+from functools import lru_cache
+from sympy import sympify as _orig_sympify
+from public import public
+
+
+@public
+@lru_cache(maxsize=256, typed=True)
+def sympify(
+ a, locals=None, convert_xor=True, strict=False, rational=False, evaluate=None
+):
+ return _orig_sympify(a, locals, convert_xor, strict, rational, evaluate)