编写设备函数

CUDA 内置目标弃用通知

Numba 内置的 CUDA 目标已弃用，进一步的开发已移至 NVIDIA numba-cuda 包。请参阅 内置 CUDA 目标弃用和维护状态。

CUDA 设备函数只能在设备内部（通过内核或另一个设备函数）调用。要定义一个设备函数

from numba import cuda

@cuda.jit(device=True)
def a_device_function(a, b):
    return a + b

与内核函数不同，设备函数可以像普通函数一样返回值。