在CUDA/CPU上執行代碼

Question

有沒有什麼辦法可以在__host__ __device__找到在CUDA或CPU上執行的功能代碼？

在此先感謝。

Answer 1

我解釋你的問題：

有沒有辦法在一個__host__ __device__函數知道是否該代碼它被在設備上或在主機上執行？

截至今天沒有運行正式可靠招檢測到它。反正你可以使用__CUDACC__宏「專業」你的設備和/或主機代碼：

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

參考：http://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#host