numba_dpex.kernel_api.private_array

Implements a simple array intended to be used inside kernel work item. Implementation is intended to be used in pure Python code when prototyping a kernel function.

Overview

Classes

PrivateArray

An array that gets allocated on the private memory of a work-item.

Classes

class PrivateArray(shape, dtype, fill_zeros=False)

An array that gets allocated on the private memory of a work-item.

The class should be used to allocate small arrays on the private per-work-item memory for fast accesses inside a kernel. It is similar in intent to the sycl::private_memory class but is not a direct analogue.