numba_extras¶
Extensions to numba
-
xarray_extras.numba_extras.
guvectorize
(signature, layout, **kwds)¶ 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()
- kwds – 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
Discussing upstream fix; see https://github.com/numba/numba/issues/2936.