-
Notifications
You must be signed in to change notification settings - Fork 155
Open
Labels
Description
Description
#1807 enabled this functionality for lapack functions. We should do it also for scalar functions (like psi), defined here:
pytensor/pytensor/link/numba/dispatch/scalar.py
Lines 75 to 77 in 7cd524c
| scalar_func_numba = wrap_cython_function( | |
| cython_func, output_dtype, input_dtypes | |
| ) |