Small change to numba interface

CalebBell · Jul 18, 2021 · acfd9a0 · acfd9a0
1 parent 677aef4
commit acfd9a0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/fluids/numba.py b/fluids/numba.py
@@ -54,6 +54,7 @@
 try:
     import scipy.special as sc
     name_to_numba_signatures = {
+        'expi': [(float64,)],
         'ellipe': [(float64,)],
         'iv': [(float64, float64,)],
         'gamma': [(float64,)],
@@ -70,6 +71,7 @@
     }
 
     name_and_types_to_pointer = {
+        ('expi', float64): ctypes.CFUNCTYPE(ctypes.c_double, ctypes.c_double)(get_cython_function_address('scipy.special.cython_special', '__pyx_fuse_1expi')),
         ('ellipe', float64): ctypes.CFUNCTYPE(ctypes.c_double, ctypes.c_double)(get_cython_function_address('scipy.special.cython_special', 'ellipe')),
         ('iv', float64, float64): ctypes.CFUNCTYPE(ctypes.c_double, ctypes.c_double, ctypes.c_double)(get_cython_function_address('scipy.special.cython_special', '__pyx_fuse_1iv')),
         ('gamma', float64): ctypes.CFUNCTYPE(ctypes.c_double, ctypes.c_double)(get_cython_function_address('scipy.special.cython_special', '__pyx_fuse_1gamma')),
@@ -84,6 +86,7 @@
         ('ellipeinc', float64, float64): ctypes.CFUNCTYPE(ctypes.c_double, ctypes.c_double, ctypes.c_double)(get_cython_function_address('scipy.special.cython_special', 'ellipeinc')),
         ('erf', float64): ctypes.CFUNCTYPE(ctypes.c_double, ctypes.c_double)(get_cython_function_address('scipy.special.cython_special', '__pyx_fuse_1erf')),
     }
+
 
     def select_kernel(name, signature):
         f2 = name_and_types_to_pointer[(name, *signature)]