gpt4 book ai didi

cudamemcpy 函数用法

转载 作者:行者123 更新时间:2023-12-01 01:08:52 47 4
gpt4 key购买 nike

在这种情况下,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;

最佳答案

在主机和设备之间复制数据时,处理双下标 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 .

关于cudamemcpy 函数用法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/16599501/

47 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com