numba_extras

Extensions to numba

xarray_extras.numba_extras.guvectorize(signature: str, layout: str, **kwargs: Any) → Callable[[Callable], Any]

Convenience wrapper around numba.guvectorize(). Generate signature for all possible data types and set a few healthy defaults.

Parameters:

signature (str) – numba signature, containing {T}
layout (str) – as in numba.guvectorize()
kwargs – passed verbatim to numba.guvectorize(). This function changes the default for cache from False to True.

example:

guvectorize("{T}[:], {T}[:]", "(i)->(i)")

Is the same as:

numba.guvectorize([
    "float32[:], float32[:]",
    "float64[:], float64[:]",
    ...
], "(i)->(i)", cache=True)

Note