Removed recompile from fastmath

Author: seanlaw

stumpy/cache.py

@@ -218,6 +218,7 @@ def _save():
     None
     """
     _enable()
+    _clear()
     _recompile()
 
     return

stumpy/fastmath.py

@@ -1,4 +1,5 @@
 import importlib
+import warnings
 
 import numba
 from numba import njit
@@ -55,11 +56,13 @@ def _set(module_name, func_name, flag):
     func = getattr(module, func_name)
     try:
         func.targetoptions["fastmath"] = flag
-        func.recompile()
+        msg = "One or more fastmath flags have been set/reset. "
+        msg += "Please call `cache._recompile()` to ensure that all njit functions "
+        msg += "are properly recompiled."
+        warnings.warn(msg)
     except AttributeError as e:
         if numba.config.DISABLE_JIT and (
             str(e) == "'function' object has no attribute 'targetoptions'"
-            or str(e) == "'function' object has no attribute 'recompile'"
         ):
             pass
         else:  # pragma: no cover

tests/test_fastmath.py

@@ -1,7 +1,7 @@
 import numba
 import numpy as np
 
-from stumpy import fastmath
+from stumpy import cache, fastmath
 
 
 def test_set():
@@ -11,11 +11,13 @@ def test_set():
 
     # case1: flag=False
     fastmath._set("fastmath", "_add_assoc", flag=False)
+    cache._recompile()
     out = fastmath._add_assoc(0, np.inf)
     assert np.isnan(out)
 
     # case2: flag={'reassoc', 'nsz'}
     fastmath._set("fastmath", "_add_assoc", flag={"reassoc", "nsz"})
+    cache._recompile()
     out = fastmath._add_assoc(0, np.inf)
     if numba.config.DISABLE_JIT:
         assert np.isnan(out)
@@ -24,11 +26,13 @@ def test_set():
 
     # case3: flag={'reassoc'}
     fastmath._set("fastmath", "_add_assoc", flag={"reassoc"})
+    cache._recompile()
     out = fastmath._add_assoc(0, np.inf)
     assert np.isnan(out)
 
     # case4: flag={'nsz'}
     fastmath._set("fastmath", "_add_assoc", flag={"nsz"})
+    cache._recompile()
     out = fastmath._add_assoc(0, np.inf)
     assert np.isnan(out)
 
@@ -39,7 +43,9 @@ def test_reset():
     # https://numba.pydata.org/numba-doc/dev/user/performance-tips.html#fastmath
     # and then reset it to the default value, i.e. `True`
     fastmath._set("fastmath", "_add_assoc", False)
+    cache._recompile()
     fastmath._reset("fastmath", "_add_assoc")
+    cache._recompile()
     if numba.config.DISABLE_JIT:
         assert np.isnan(fastmath._add_assoc(0.0, np.inf))
     else:  # pragma: no cover