pttools.speedup.jit

Custom decorators for JIT-compilation with Numba

Warning: complex decorators calling Numba may cause segmentation faults when profiled. https://github.com/numba/numba/issues/3625

Functions

pttools.speedup.jit.njit(func=None, **kwargs)

Wrapper for numba.njit.

May cause segmentation faults with profilers.

Parameters:

func (callable)

pttools.speedup.jit.njit_if_numba_integrate(func=None, **kwargs)

Parameters:

func (callable)

Return type:

callable

pttools.speedup.jit.njit_module(**kwargs)

Adapted from numba.jit_module.

May cause segmentation faults with profilers.

pttools.speedup.jit.vectorize(**kwargs)

Extended version of numba.vectorize with support for NUMBA_DISABLE_JIT