cudaMemcpy2D 的分段错误

Question

我有一个存储在 GPU 中的 2D 数组 dev_histogram 和一个存储在 CPU 中的 2D 数组 histogarm 。我想将 dev_histogram 的内容复制到直方图中。以下是我的程序的相关部分。我也可以发布完整的代码。

int *dev_histogram; // Array for histogram, GPU
int histogram[SIZE_THETA][SIZE_RHO]; // Array for histogram, CPU

size_t pitch;
histogramSize = sizeof(int) * SIZE_THETA * SIZE_RHO;
cudaMallocPitch((void**)&dev_histogram, &pitch, SIZE_THETA * sizeof(int), SIZE_RHO)

houghTransformation << <width, height >> >(dev_edges, dev_histogram, pitch, n_pixels, width, height);

// Here I get a Segmentation fault:
cudaMemcpy2D(histogram, pitch, dev_histogram, SIZE_THETA * sizeof(int), SIZE_THETA * sizeof(int), SIZE_RHO * sizeof(int), cudaMemcpyDeviceToHost)

您能帮我了解如何将矩阵复制回来吗？大多数情况下，我对如何作为我的来源的宣传感到困惑。

Answer 1

指定 SIZE_RHO 作为高度，而不是 SIZE_RHO * sizeof(int)：

<cudaMemcpy2D(histogram, pitch, dev_histogram, SIZE_THETA * sizeof(int), SIZE_THETA * sizeof(int), SIZE_RHO * sizeof(int), cudaMemcpyDeviceToHost);
>cudaMemcpy2D(histogram, pitch, dev_histogram, SIZE_THETA * sizeof(int), SIZE_THETA * sizeof(int), SIZE_RHO, cudaMemcpyDeviceToHost);

Answer 2

在 CUDA 工具包参考手册中，您可以看到 cudaMallocPitch 中的间距是为要复制的 2D 数组分配的宽度（以字节为单位）。您的 dev_histogram 将具有等于间距的实际宽度和等于您指定的高度的高度。 2D 数组的每一行都分配有间距字节，但只有 width*sizeof(int) 字节的有效数据。

在同一文档中，cudaMemcpy2D 的原型位于

cudaError_t cudaMemcpy2D (void ∗ dst, size_t dpitch, const void ∗ src, size_t spitch, size_t width, size_t height, enum cudaMemcpyKind kind)

此处 dst 是主机上的数组，dpitch 是目标数组（直方图）的字节宽度，spitch 是源数组（dev_histogram）的字节宽度。宽度和高度是二维数组的尺寸。
那么你必须这样称呼它：

cudaMemcpy2D(histogram, SIZE_THETA*sizeof(int), dev_histogram, pitch, SIZE_THETA * sizeof(int), SIZE_RHO, cudaMemcpyDeviceToHost);

编辑：在ArchaeaSoftware之后我注意到高度实际上是行数，字节数的高度没有意义。更新了答案，因为您仍然需要更改音调。

Answer 3

通常，当在连续内存中存储数据时，您希望使内存部分的维度为存储单元的倍数，以便可以有效地读取数据。例如，您可以读取一个 32 位字，而不是连续读取 4 个单独的字节。你这样做是为了效率。查找内存对齐方式。

出于同样的原因，您希望使某些数组的大小为节距*高度，其中节距是向上舍入到您正在使用的任何存储单元的最接近倍数的宽度。如果您的数组为 31*5，则使用 32 的间距和 31 的宽度。 4 个 32 位读取预计将比 31 个 1 字节读取更快。您丢弃额外的“填充”字节。

您可能想设置间距=宽度。你的段错误的原因是你没有初始化它。检查宽度和高度是否与您的 GPU 线程块大小规格兼容。