Example overloading a method in Numba's CUDA target

overload_method_example.py

from numba.extending import overload_method
from numba import types

@overload_method(types.Array, 'sum', target='cuda')
def array_sum(arr):
    if arr.ndim != 1:
        return None

    def sum_impl(arr):
        res = 0 
        for i in range(len(arr)):
            res += arr[i]
        return res 
    return sum_impl