c++ - 如何在 cpp 和 cuda 模块中使用 cuda 类头文件

Question

我正在尝试在我的设备中创建 Color 对象。这是我所拥有的简化版本:

在 Color.hpp 中:

class Color { 
public:
    Color(){}
    float r, g, b;
    // other functions
}

在 test.cu 中:

__global__ void runCuda(){
    Color c = Color();
}

int main() {
   runCuda<<<1,1>>>()
}

这给了我一个错误说

calling a host function from a global function is not allowed

所以这很好。我只需要在 Color(){} 函数前面添加 __host__ 和 __device__ 。

但随后出现以下错误:

host does not name a type

据我所知，这是因为我没有使用 nvcc 编译它。问题是我正在使用 CMake 来构建我的项目。我不太确定它是如何做到的，但它似乎是用 c++ 编译器编译 .cpp 文件，用 nvcc 编译器编译 .cu 文件。

但在我的设备中，我想创建 Color 对象。有没有办法在我的 CMakefile 或我的代码中解决这个问题？或者我是否需要为所有现有类创建一个 cuda 版本？

Answer 1

像这样的东西应该可以工作:

#ifdef __CUDACC__
#define CUDA_HOSTDEV __host__ __device__
#else
#define CUDA_HOSTDEV
#endif

class Color { 
public:
    CUDA_HOSTDEV Color(){}
    float r, g, b;
    // other functions
};

当您的 hpp 文件包含在 .cu 文件中时，__CUDACC__ macro将被定义。当它包含在 .cpp 文件中时(nvcc 将其交给主机编译器)，宏将不会被定义。