如何创建一个可以从主机和设备调用的内核函数？

Question

以下试验提出了我的意图,但未能编译:

__host__ __device__ void f(){}

int main()
{
    f<<<1,1>>>();
}

编译器投诉:

a.cu(5): error: a __device__ function call cannot be configured

1 error detected in the compilation of "/tmp/tmpxft_00001537_00000000-6_a.cpp1.ii".

希望我的陈述清楚,并感谢您的建议.

Answer 1

您需要创建一个CUDA内核入口点,例如__global__function.就像是:

#include <stdio.h>

__host__ __device__ void f() {
#ifdef __CUDA_ARCH__
    printf ("Device Thread %d\n", threadIdx.x);
#else
    printf ("Host code!\n");
#endif
}

__global__ void kernel() {
   f();
}

int main() {
   kernel<<<1,1>>>();
   if (cudaDeviceSynchronize() != cudaSuccess) {
       fprintf (stderr, "Cuda call failed\n");
   }
   f();
   return 0;
}