CUDA 与 CPU 和 GPU 的功能相同

CUDA same function for CPU and GPU

本文关键字:功能 GPU CPU CUDA      更新时间:2023-10-16

为了从主机代码和GPU内核调用相同的函数,我是否必须保留相同函数的两个副本,如下所示:

int sum(int a, int b){
return a+b;
}
__device int sumGPU(int a, int b){
return a+b;
}

或者是否有任何技术可以保留/管理函数的单个副本?

您只需添加 __host__ 关键字即可从主机或设备调用函数。

__host__ __device__ int sum(int a, int b){
  return a+b;
}