Open rajeeja opened 2 hours ago
custom decorator
# Custom decorator to enable/disable numba
def optional_numba_jit(nopython=True, parallel=False, cache=False, fastmath=False):
def decorator(func):
if USE_NUMBA:
return numba.jit(nopython=nopython, parallel=parallel, cache=cache, fastmath=fastmath)(func)
else:
return func
return decorator
Proposed new feature or change:
codecov doesn't work correctly and track @njit decorated functions, a way to enable disable numba across the project would be useful.