Function CUDA/CPU上的代码执行

有没有办法在

\uuuuuu主机\uuuuuuuuuuuuuuuuuuuuuu设备\uuuuuuuuuu

功能中找到在CUDA或CPU上执行的代码

---

提前谢谢。

我把你的问题解释为：

在

\uuuuuuu主机\uuuuuu设备\uuuuuu

功能中，是否有任何方法可以知道它是在设备上还是在主机上执行的代码

到今天为止，还没有正式可靠的运行时技巧来检测它。无论如何，您可以使用

\uuuu CUDACC\uuuu

宏“专门化”设备和/或主机的代码：

__host__ __device__ void yourfunction() {
  #ifdef __CUDACC__
     // Device code branch
  #else
     // Host code branch
  #endif
}

参考：