numba_dpex.core#

Subpackages#

Overview#

Classes#
`DpctlSyclQueue`	A Numba type to represent a dpctl.SyclQueue PyObject.
`DpctlSyclEvent`	A Numba type to represent a dpctl.SyclEvent PyObject.
`DpnpNdArray`	The Numba type to represent an dpnp.ndarray. The type has the same
`IntEnumLiteral`	A Literal type for IntEnum objects. The type contains the original Python
`KernelDispatcherType`	The type of KernelDispatcher dispatchers
`NdRangeType`	Numba-dpex type corresponding to
`RangeType`	Numba-dpex type corresponding to
`USMNdArray`	A type class to represent dpctl.tensor.usm_ndarray.

Attributes#
`none`	-
`uint32`	-
`uint64`	-
`int32`	-
`int64`	-
`float32`	-
`float64`	-
`b1`	-
`i4`	-
`i8`	-
`u4`	-
`u8`	-
`f4`	-
`f8`	-
`float_`	-
`double`	-
`usm_ndarray`	-
`void`	-

Classes#

class DpctlSyclQueue(sycl_queue)#

Bases: numba.types.Type

A Numba type to represent a dpctl.SyclQueue PyObject.

Overview

Methods#
`rand_digit_str`(n)	-

Members

rand_digit_str(n)#

class DpctlSyclEvent#

Bases: numba.types.Type

A Numba type to represent a dpctl.SyclEvent PyObject.

class DpnpNdArray(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_dpex.core.types.usm_ndarray_type.USMNdArray

The Numba type to represent an dpnp.ndarray. The type has the same structure as USMNdArray used to represent dpctl.tensor.usm_ndarray.

class IntEnumLiteral(value)#

Bases: numba.core.types.Literal, numba.core.types.Integer

A Literal type for IntEnum objects. The type contains the original Python value of the IntEnum class in it.

Overview

Methods#
`can_convert_to`(typingctx, other)	Check whether this type can be converted to the other.

Members

can_convert_to(typingctx, other) → bool#: Check whether this type can be converted to the other. If successful, must return a string describing the conversion, e.g. “exact”, “promote”, “unsafe”, “safe”; otherwise None is returned.

class KernelDispatcherType(dispatcher)#

Bases: numba.core.types.Dispatcher

The type of KernelDispatcher dispatchers

Overview

Methods#
`cast_python_value`(args)	abc -

Members

abstract cast_python_value(args)#

class NdRangeType(ndim: int)#

Bases: numba.core.types.Type

Numba-dpex type corresponding to numba_dpex.kernel_api.ranges.NdRange

class RangeType(ndim: int)#

Bases: numba.core.types.Type

Numba-dpex type corresponding to numba_dpex.kernel_api.ranges.Range

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.

Attributes#

none#

uint32#

uint64#

int32#

int64#

float32#

float64#

b1#

i4#

i8#

u4#

u8#

f4#

f8#

float_#

double#

usm_ndarray#

void#