cuda - 不同CUDA版本的不同构建规则？

Question

我想在我的 CUDA 代码中指定我是使用 CUDA 构建规则版本 4.2 还是 CUDA 构建规则版本 5 进行编译。

是否有用于此目的的宏？

注意：我知道有一个选项可以指定当前编译过程是否支持某些架构，例如__CUDA_ARCH__.

我正在为 CUDA 构建规则版本寻找类似的宏。

Answer 1

用于确定 CUDA Toolkit 版本的宏是CUDART_VERSION

您可以根据 CUDA Toolkit 版本对代码进行条件编译，如下所示：

#if CUDART_VERSION == 5000

//Compilation is being done using CUDA Toolkit 5.0

#elif CUDART_VERSION == 4200

//Compilation is being done using CUDA Toolkit 4.2
.
.
.
//And so on
#else

#endif