对 CUDA 代码使用 C++ 头结构

Question

我知道可能有人问过这个问题，但我找不到特别能回答我的问题的答案，所以就这样吧……

问题很简单，我正在尝试为 CUDA (.cu/.cuh) 代码使用 C++ 风格的头文件。

//MyClass.cuh
#ifndef MY_CLASS
#define MY_CLASS
#include ...cuda.h, etc.
class MyClass
{
   public:
      void myFunction();   
   private:
      __global__ void myKernel();
}
#endif

//MyClass.cu
#include "MyClass.cuh"
void MyClass::myFunction()
{
   //myFunction definition...
}

__global__ void MyClass::myKernel()
{
   //myKernel definition...
}

这会起作用吗？

Answer 1

不，这行不通：

class MyClass
{
    ...
      __global__ void myKernel();

编译器不会让你定义一个__global__函数的类成员函数（试试看）。 即使可以，在设备上运行的此类函数也不具有对其他类数据或函数成员的通常类访问权限。 所以通常的建议是在类的范围之外声明你的内核定义，并在调用内核的类中有一个包装成员函数。

一些额外的讨论在这里和这里