cuda - 不同架构的不同内核

Question

我想知道是否有一些简单的方法可以为不同的体系结构拥有不同版本的内核。他们是一个简单的方法吗？或者唯一的可能性是在独立文件中定义独立内核并要求 nvcc 编译为每个文件的不同架构？

Answer 1

您可以通过编译器指令来做到这一点。就像是

__global__ void kernel(...) {

# if __CUDA_ARCH__ >= 350

    do something

# else

    do something else

# endif

}