c++ - CUDA 从单独的文件调用设备函数(名称修改？)

Question

我应该如何正确执行此操作？这是代码的简化:

//main.cu    
#include "math.cuh"

__global__ void test(float *x, unsigned numElements)
{
    int i = blockDim.x * blockIdx.x + threadIdx.x;

    if (i < numElements)
    {
        float array[5] = {1, 2, 3, 4, 5};
        copyArray(x + 5*i, array, 5);
    }
}

int main(int argc, char **argv)
{
    test<<<blocksPerGrid, threadsPerBlock>>>(d_A, numElements);
}

//math.cuh
__device__ void copyArray(float *dest, float *src, unsigned length);

//math.cu
#include "math.cuh"
__device__ void copyArray(float *dest, float *src, size_t length)
{
    for (int i = 0; i < length; i++) {
        dest[i] = src[i];
    }
}

用这个命令编译:

nvcc -rdc=true -arch=sm_20 -o cudaMain main.cu math.cu -Xlinker -framework,OpenGL,-framework,GLUT && ./cudaMain

得到这个错误:

nvlink error   : Undefined reference to '_Z9copyArrayPfS_j' in '/tmp/tmpxft_00000265_00000000-21_main.o'

这显然看起来像一个名称修改错误，但我尝试将 extern "C"放在各种地方，但它没有用。

Answer 1

函数原型(prototype)使用unsigned，而定义使用size_t。是这个原因吗？