CUDA: Allow dynamic allocation of local arrays
Author: gmarkallCreated Dec 4, 2020Updated Sep 6, 2026
Labelsfeature_requestCUDAgood second issue
Dynamic allocation is supported on CC 2.0 and above devices, so it would be nice if cuda.local.array shape could be variable instead of constant, e.g.:
@cuda.jit
def kernel(nx, ny):
arr = cuda.local.array((nx, ny), dtype=dt)Source: numba/numba