使用C ++模板分配CUDA设备内存时发出警告

Question

我声明了以下模板以使代码更短：

template <typename T>
void allocateGPUSpace(T* ptr, int size){

    cudaMalloc((void**)&ptr,size * sizeof(T));
}

此外，我使用模板如下：

float* alphaWiMinusOne;

allocateGPUSpace<float>( alphaWiMinusOne,numUnigrams);

但是，当我编译代码时，VS 2008发出警告

 warning: variable "alphaWiMinusOne" is used before its value is set

和

uninitialized local variable 'alphaWiMinusOne' used

cuda不理解C ++中的模板吗？ 天哪，这对于英伟达来说是必须要做的

Answer 1

首先，该警告不是来自CUDA，而是来自主机编译器（因此Microsoft的C ++编译器或GCC取决于您的平台），并且它是完全有效的警告。 您在这里犯了同样的错误，并且此代码无法按您希望的那样工作，因为您正在传递指针以按值而不是按引用进行操作。 您的代码应如下所示：

template <typename T> 
void allocateGPUSpace(T ** ptr, int size){   
    cudaMalloc((void**)ptr, size * sizeof(T)); 
} 

和这样的电话：

float * alphaWiMinusOne;
allocateGPUSpace<float>(&alphaWiMinusOne, numUnigrams);

也许

template <typename T> 
T * allocateGPUSpace(int size){
    T * ptr;   
    cudaMalloc((void**)&ptr, size * sizeof(T));
    return ptr;
} 

接着

float * alphaWiMinusOne = allocateGPUSpace<float>(numUnigrams);

使用任何一种都将消除编译器警告，并且代码将起作用。 作为样式的注释，这将是一个短视的辅助函数设计，不包含任何错误检查。