numba_dpex.numpy_usm_shared module
- class numba_dpex.numpy_usm_shared.DparrayAsNdarray(context)
Bases:
numba.core.typing.templates.CallableTemplate- generic()
- class numba_dpex.numpy_usm_shared.DparrayFromNdarray(context)
Bases:
numba.core.typing.templates.CallableTemplate- generic()
- class numba_dpex.numpy_usm_shared.UsmArrayAttribute(context)
Bases:
numba.core.typing.templates.AttributeTemplate- convert_array_to_usmarray(retty)
- generic_resolve(ary, attr)
- key
- resolve_T(ary)
- resolve_argsort(ary, args, kws)
- resolve_astype(ary, args, kws)
- resolve_copy(ary, args, kws)
- resolve_ctypes(ary)
- resolve_dtype(ary)
- resolve_flags(ary)
- resolve_flat(ary)
- resolve_flatten(ary, args, kws)
- resolve_imag(ary)
- resolve_item(ary, args, kws)
- resolve_itemset(ary, args, kws)
- resolve_itemsize(ary)
- resolve_ndim(ary)
- resolve_nonzero(ary, args, kws)
- resolve_ravel(ary, args, kws)
- resolve_real(ary)
- resolve_reshape(ary, args, kws)
- resolve_shape(ary)
- resolve_size(ary)
- resolve_sort(ary, args, kws)
- resolve_strides(ary)
- resolve_take(ary, args, kws)
- resolve_transpose(ary, args, kws)
- resolve_view(ary, args, kws)
- class numba_dpex.numpy_usm_shared.UsmSharedArrayType(*args, **kwargs)
Bases:
numba_dpex.dppy_array_type.DPPYArrayCreates a Numba type for Numpy arrays that are stored in USM shared memory. We inherit from Numba’s existing Numpy array type but overload how this type is printed during dumping of typing information and we implement the special __array_ufunc__ function to determine who this type gets combined with scalars and regular Numpy types. We re-use Numpy functions as well but those are going to return Numpy arrays allocated in USM and we use the overloaded copy function to convert such USM-backed Numpy arrays into typed USM arrays.
- property box_type
Returns the Python type to box to.
- copy(*args, **kwargs)
- numba_dpex.numpy_usm_shared.argspec_to_string(argspec)
- numba_dpex.numpy_usm_shared.box_array(typ, val, c)
- numba_dpex.numpy_usm_shared.copy_func_for_usmarray(f, usmarray_mod)
- numba_dpex.numpy_usm_shared.dprint(*args)
- numba_dpex.numpy_usm_shared.is_usm_callback(obj)
- numba_dpex.numpy_usm_shared.numba_register()
- numba_dpex.numpy_usm_shared.numba_register_lower_builtin()
- numba_dpex.numpy_usm_shared.numba_register_typing()
- numba_dpex.numpy_usm_shared.typeof_ta_ndarray(val, c)
- numba_dpex.numpy_usm_shared.types_replace_array(x)
- numba_dpex.numpy_usm_shared.usmarray_conversion_as(context, builder, sig, args)
- numba_dpex.numpy_usm_shared.usmarray_conversion_from(context, builder, sig, args)