Cudamemcpy 函数使用

Question

cudaMemcpy 函数在这种情况下如何工作？

我已经声明了一个这样的矩阵

float imagen[par->N][par->M];

我想将它复制到 cuda 设备，所以我这样做了

float *imagen_cuda;

int tam_cuda=par->M*par->N*sizeof(float);

cudaMalloc((void**) &imagen_cuda,tam_cuda); 
cudaMemcpy(imagen_cuda,imagen,tam_cuda,cudaMemcpyHostToDevice);

这会将二维数组复制到一维数组中吗？

我怎样才能复制到另一个二维数组？我可以更改它吗？它会起作用吗？

float **imagen_cuda;

Answer 1

在主机和设备之间复制数据时，处理双下标 C 数组并非易事。在大多数情况下，cudaMemcpy（包括cudaMemcpy2D）需要一个用于源和目标的普通指针，而不是指向指针的指针。

（我认为）最简单的方法是在主机和设备上“展平”二维数组，并使用索引算法来模拟二维坐标：

float imagen[par->N][par->M];
float *myimagen = &(imagen[0][0]);
float myval = myimagen[(rowsize*row) + col];

然后您可以使用普通的 cudaMemcpy 操作来处理传输（使用 myimagen 指针）：

float *d_myimagen;
cudaMalloc((void **)&d_myimagen, (par->N * par->M)*sizeof(float));
cudaMemcpy(d_myimagen, myimagen, (par->N * par->M)*sizeof(float), cudaMemcpyHostToDevice);

如果您真的想处理动态大小（即在编译时未知）双下标数组，您可以查看此question/answer。