multidimensional-array - 在CUDA中的设备内存上分配2D阵列

Question

如何在Cuda的设备内存中分配和传输2D阵列（与主机之间）？

Answer 1

我找到了解决该问题的方法。我不必弄平阵列。

内置的cudaMallocPitch()功能可以完成这项工作。我可以使用cudaMemcpy2D()函数在设备之间来回传输阵列。

例如

cudaMallocPitch((void**) &array, &pitch, a*sizeof(float), b);

这将创建一个尺寸为a * b的2D数组，其间距作为参数传入。

以下代码创建2D数组并在元素上循环。它易于编译，您可以使用它。

#include<stdio.h>
#include<cuda.h>
#define height 50
#define width 50

// Device code
__global__ void kernel(float* devPtr, int pitch)
{
    for (int r = 0; r < height; ++r) {
        float* row = (float*)((char*)devPtr + r * pitch);
        for (int c = 0; c < width; ++c) {
             float element = row[c];
        }
    }
}

//Host Code
int main()
{

float* devPtr;
size_t pitch;
cudaMallocPitch((void**)&devPtr, &pitch, width * sizeof(float), height);
kernel<<<100, 512>>>(devPtr, pitch);
return 0;
}