2

我需要转置一个方阵。我用矩阵测试程序:a[i][j] = 0 if i>j, a[i][j] = if i<=j,但结果表明并非所有元素都在正确的位置。

这是代码(main() 除外):

#include <stdio.h> 
#include <stdlib.h>
__global__ void transpose_kernel (float *a, float *b, int n) {
    unsigned int ax = blockDim.x * blockIdx.x + threadIdx.x;
    unsigned int ay = blockDim.y * blockIdx.y + threadIdx.y;
    unsigned int aIdx = ax + n * ay;
    unsigned int bIdx = ay + n * ax;

    b[bIdx] = a[aIdx];
}

int transpose_host (float *a, float *b, int n) {
    int size = n * n * sizeof (float);
    float *aDev = NULL, *bDev = NULL;

    cudaError_t cuerr = cudaMalloc ((void**)&aDev, size);
    if (cuerr != cudaSuccess) {
        fprintf (stderr, "Cannot allocate GPU memory for aDev: %s\n", cudaGetErrorString (cuerr));
        return (-1);
    }

cuerr = cudaMalloc ((void**)&bDev, size);
if (cuerr != cudaSuccess) {
    fprintf (stderr, "Cannot allocate GPU memory for bDev: %s\n", cudaGetErrorString (cuerr));
    return (-1);
}

dim3 blockSize = dim3 (16, 16, 1);
dim3 gridSize = dim3 (n/16 + 1, n/16 + 1, 1);

cuerr = cudaMemcpy (aDev, a, size, cudaMemcpyHostToDevice);
if (cuerr != cudaSuccess) {
    fprintf (stderr, "Cannot copy data from a to aDev: %s\n", cudaGetErrorString (cuerr));
    return (-1);
}

transpose_kernel <<< gridSize, blockSize >>> (aDev, bDev, n);

cuerr = cudaGetLastError ();
if (cuerr != cudaSuccess) {
    fprintf (stderr, "Cannot launch CUDA kernel: %s\n", cudaGetErrorString (cuerr));
    return (-1);
}

cuerr = cudaDeviceSynchronize ();
if (cuerr != cudaSuccess) {
    fprintf (stderr, "Cannot synchronize CUDA kernel: %s\n", cudaGetErrorString (cuerr));
    return (-1);
}

cuerr = cudaMemcpy (b, bDev, size, cudaMemcpyDeviceToHost);
if (cuerr != cudaSuccess) {
    fprintf (stderr, "Cannot copy data from b to bDev: %s\n", cudaGetErrorString (cuerr));
    return (-1);
}

cudaFree (aDev);
cudaFree (bDev);

    return (0);
}

为什么我的数组转置不正确?

4

1 回答 1

2

问题出在分配数组之外的“额外”线程中。

当你分配你的网格块时,你会四舍五入(事实上,即使事情平分,也会强制四舍五入到下一个整数:)

dim3 gridSize = dim3 (n/16 + 1, n/16 + 1, 1);

这样总有线程的 ax 或 ay 落在 [0,n) 之外。因此,当您无论如何复制时,您就是将随机数据复制a[aIdx]b[bIdx]内存中,实际上可能会根据调度覆盖“真实”数据。

您可以通过更改内核来检查此问题来解决此问题:

if (ax < n && ay < n)
    b[bIdx] = a[aIdx];

如果事情平均分配,您可能希望将网格大小的四舍五入更改为不四舍五入:

dim3 gridSize = dim3 ((n+15)/16, (n+15)/16, 1);
于 2012-11-09T21:01:18.053 回答