ThinkChat2.0新版上线,更智能更精彩,支持会话、画图、阅读、搜索等,送10W Token,即刻开启你的AI之旅 广告
# 3.4。编写设备功能 > 原文: [http://numba.pydata.org/numba-doc/latest/cuda/device-functions.html](http://numba.pydata.org/numba-doc/latest/cuda/device-functions.html) CUDA 设备功能只能从设备内部调用(通过内核或其他设备功能)。要定义设备功能: ```py from numba import cuda @cuda.jit(device=True) def a_device_function(a, b): return a + b ``` 与内核函数不同,设备函数可以返回与普通函数类似的值。