numba_extras — xarray_extras 0.6.1.dev7+g2fd6faf.d20250305 documentation (original) (raw)

• numba_extras
• View page source

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)