Numba 0.56 does not support `atomic.add` on arrays of complexs anymore

Reporting a bug

• I have tried using the latest released version of Numba (most recent is visible in the change log (https://github.com/numba/numba/blob/main/CHANGE_LOG).
• I have included a self contained code sample to reproduce the problem. i.e. it’s possible to run as ‘python bug.py’.

The following code snippet uses an atomic.add on an array of complex numbers. It runs without any issues with numba 0.55.2, but fails with numba 0.56

from numba import cuda
import numpy as np

@cuda.jit
def atomic_add_one(values):
    """
    Adds 1 to each element values, with an atomic add

    Parameters:
    -----------
    values: 1darray of `complex128`
    """
    tid = cuda.threadIdx.x
    bid = cuda.blockIdx.x
    bdim = cuda.blockDim.x
    i = (bid * bdim) + tid
    cuda.atomic.add(values.real, i, 1)

arr = np.random.rand(16384) + 1.j*np.random.rand(16384)
print(arr.dtype)

atomic_add_one[256,64](arr)

[gmarkall]

One potential fix is now ready for consideration in #8310. Whether this is the right fix and the general path forward is a complex decision, so I’ve added it to the agenda for next week’s dev meeting along with the considerations I can see as of now: https://hackmd.io/P1y0q3BcStKfIZky7CLHHg?view

[gmarkall]

Thanks for the reproducer - I can reproduce this issue, and I’m looking into it.