lemke_howson: Add cache=True

QuantEcon · Nov 14, 2016 · 2c2f54b · 2c2f54b
1 parent 918c687
commit 2c2f54b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/quantecon/game_theory/lemke_howson.py b/quantecon/game_theory/lemke_howson.py
@@ -129,7 +129,7 @@ def lemke_howson(g, init_pivot=0, max_iter=10**6, full_output=False):
     return NE, res
 
 
-@jit(nopython=True)
+@jit(nopython=True, cache=True)
 def initialize_tableaux(payoff_matrices, tableaux, bases):
     """
     Given a tuple of payoff matrices, initialize the tableau and basis
@@ -224,7 +224,7 @@ def initialize_tableaux(payoff_matrices, tableaux, bases):
     return tableaux, bases
 
 
-@jit(nopython=True)
+@jit(nopython=True, cache=True)
 def lemke_howson_tbl(tableaux, bases, init_pivot, max_iter):
     """
     Main body of the Lemke-Howson algorithm implementation.
@@ -327,7 +327,7 @@ def lemke_howson_tbl(tableaux, bases, init_pivot, max_iter):
     return converged, num_iter
 
 
-@jit(nopython=True)
+@jit(nopython=True, cache=True)
 def pivoting(tableau, pivot, pivot_row):
     """
     Perform a pivoting step. Modify `tableau` in place.
@@ -367,7 +367,7 @@ def pivoting(tableau, pivot, pivot_row):
     return tableau
 
 
-@jit(nopython=True)
+@jit(nopython=True, cache=True)
 def min_ratio_test_no_tie_breaking(tableau, pivot, test_col,
                                    argmins, num_candidates):
     """
@@ -420,7 +420,7 @@ def min_ratio_test_no_tie_breaking(tableau, pivot, test_col,
     return num_argmins
 
 
-@jit(nopython=True)
+@jit(nopython=True, cache=True)
 def lex_min_ratio_test(tableau, pivot, slack_start, argmins):
     """
     Perform the lexico-minimum ratio test.
@@ -468,7 +468,7 @@ def lex_min_ratio_test(tableau, pivot, slack_start, argmins):
     return argmins[0]
 
 
-@jit(nopython=True)
+@jit(nopython=True, cache=True)
 def get_mixed_actions(tableaux, bases):
     """
     From `tableaux` and `bases`, extract non-slack basic variables and