#6549·numba

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.:

python
@cuda.jit
def kernel(nx, ny):
    arr = cuda.local.array((nx, ny), dtype=dt)