Skip to content

Commit 8cebd03

Browse files
committed
RFC: Replace @generated_jit with @overload in _compute_fp
1 parent 8e241a0 commit 8cebd03

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

Diff for: quantecon/_compute_fp.py

+8-3
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@
66
import time
77
import warnings
88
import numpy as np
9-
from numba import jit, generated_jit, types
9+
from numba import jit, types
10+
from numba.extending import overload
1011
from .game_theory.lemke_howson import _lemke_howson_tbl, _get_mixed_actions
1112

1213

@@ -352,8 +353,12 @@ def _initialize_tableaux_ig(X, Y, tableaux, bases):
352353
return tableaux, bases
353354

354355

355-
@generated_jit(nopython=True, cache=True)
356-
def _square_sum(a):
356+
def _square_sum(a): # pragma: no cover
357+
pass
358+
359+
360+
@overload(_square_sum, jit_options={'cache':True})
361+
def _square_sum_ol(a):
357362
if isinstance(a, types.Number):
358363
return lambda a: a**2
359364
elif isinstance(a, types.Array):

0 commit comments

Comments
 (0)