Skip to content

3.4。编写设备功能

原文: http://numba.pydata.org/numba-doc/latest/cuda/device-functions.html

CUDA 设备功能只能从设备内部调用(通过内核或其他设备功能)。要定义设备功能:

from numba import cuda

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

与内核函数不同,设备函数可以返回与普通函数类似的值。



回到顶部