CUDA: Add Support to Creating StructModel Array - #7772

from numba import cuda, types
from numba.core.extending import register_model, models, make_attribute_wrapper

class Float3(types.Type):
    def __init__(self):
        super().__init__(name='float3')

float3 = Float3()

@register_model(Float3)
class Float3Model(models.StructModel):
    def __init__(self, dmm, fe_type):
        members = [
            ("x", types.float32),
            ("y", types.float32),
            ("z", types.float32),
        ]
        super().__init__(dmm, fe_type, members)


make_attribute_wrapper(Float3, "x", "x")
make_attribute_wrapper(Float3, "y", "y")
make_attribute_wrapper(Float3, "z", "z")

@cuda.jit
def use_struct_array_kernel():
    x = cuda.local.array(3, dtype=float3)

use_struct_array_kernel[1, 1]()