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.