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 .