random.utilities: Cache guvectorize

QuantEcon · Dec 11, 2018 · 045168f · 045168f
1 parent d869c88
commit 045168f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/quantecon/random/utilities.py b/quantecon/random/utilities.py
@@ -89,10 +89,10 @@ def _probvec(r, out):
     out[n] = 1 - r[n-1]
 
 _probvec_parallel = guvectorize(
-    ['(f8[:], f8[:])'], '(n), (k)', nopython=True, target='parallel'
+    ['(f8[:], f8[:])'], '(n), (k)', nopython=True, target='parallel', cache=True,
     )(_probvec)
 _probvec_cpu = guvectorize(
-    ['(f8[:], f8[:])'], '(n), (k)', nopython=True, target='cpu'
+    ['(f8[:], f8[:])'], '(n), (k)', nopython=True, target='cpu', cache=True,
     )(_probvec)