訪問CUDA庫函數內的__device__函數模板專門實例

Question

我有以下模板__device__功能CUDA：

template<typename T> 
__device__ void MyatomicAdd(T *address, T val){ 
atomicAdd(address , val); 
} 

，編譯，如果有T實例化爲浮點運行良好，即

__global__ void myKernel(float *a, float b){ 
MyatomicAdd<float>(a,b); 
} 

將運行沒有問題。

我想專精這個功能，因爲雙打沒有atomicAdd()，所以我可以用雙精度手工編寫一個實現。忽略對於現在的雙精度專業化，單精度專業化和模板是這樣的：

template<typename T> 
__device__ void MyatomicAdd(T *address, T val){ 
}; 


template<> 
__device__ void MyatomicAdd<float>(float *address, float val){ 
atomicAdd(address , val); 
} 

現在，編譯器抱怨atomicAdd（）是我的專業化不確定的，這同樣適用，當我嘗試使用任何CUDA功能如專業化中的__syncthreads（）。有任何想法嗎？謝謝。

Answer 1

它最終成爲與同事開發的一些OpenGL代碼的鏈接問題。迫使專業化內聯解決問題，但顯然不是根本原因。不過，現在它會做，直到我可以被困擾挖掘其他人的代碼。