numba_dpex.core.types.usm_ndarray_type

A type class to represent dpctl.tensor.usm_ndarray type in Numba

Overview

Classes

USMNdArray

A type class to represent dpctl.tensor.usm_ndarray.

Classes

class USMNdArray(ndim, layout='C', dtype=None, usm_type='device', device=None, queue=None, readonly=False, name=None, aligned=True, addrspace=address_space.GLOBAL.value)

Bases: numba.core.types.npytypes.Array

A type class to represent dpctl.tensor.usm_ndarray.

Overview

Methods

copy(dtype, ndim, layout, readonly, addrspace, device, usm_type)	-
unify(typingctx, other)	Unify this with the other USMNdArray.
can_convert_to(typingctx, other)	Convert this USMNdArray to the other.

Members

copy(dtype=None, ndim=None, layout=None, readonly=None, addrspace=None, device=None, usm_type=None)

unify(typingctx, other)

Unify this with the other USMNdArray.

can_convert_to(typingctx, other)

Convert this USMNdArray to the other.