0

我正在尝试在 cuda 中创建一个二维数组并对其进行初始化，但失败得很惨。这是我的初始化内核：

__global__ void initMap(float* map, size_t pitch, int w, int h, int numX, int numY){
int idx=blockIdx.x*blockDim.x+threadIdx.x;
int idy=blockIdx.y*blockDim.y;
int i;
int j;
for (i=0; i<numX; i++){
    for (j=0; j<numY; j++){
        int idMC=idx+i;
        int idMR=threadIdx.y+j;
        if(idMC<w && idy+idMR<h){
             float* row=(float*)(map+idy+idMR*pitch);
             row[idMC]=0.5;
        }
    }
}

__syncthreads();
}

以下是我分配数组并在主中调用初始化内核的方式：

int width=map_size;
int height=map_size;
float* map;
size_t pitch;
checkCudaErrors(cudaMallocPitch(&map, &pitch, width*sizeof(float), height));
int numT=32;
int numBX=(int)ceil((float)width/numT);
int numBY=(int)ceil((float)height/numT);
dim3 numBlocks(numBX, numBY);
dim3 numThr(numT, numT);
initMap <<<numBlocks, numThr>>> (map, pitch/sizeof(float), width, height, 1, 1);
cudaError_t err=cudaGetLastError();
if (err != cudaSuccess) 
    printf("Error: %s\n", cudaGetErrorString(err));
checkCudaErrors(cudaDeviceSynchronize());

基本上，我要做的是将 2D 数组分成 32x32 块，并将每个块分配给一个块，然后用 0.5 填充它。但是，当我尝试将阵列写入磁盘或从另一个内核访问其元素时，我可以看到那里有很多 QNAN，所以我想我并没有填满整个阵列，而是遗漏了一些地方。

4

1 回答 1

1

像这样的东西应该工作：

__global__ void initMap(float* map, size_t pitch, int w, int h, int numX, int numY){
  int col = threadIdx.x + (blockDim.x * blockIdx.x);
  int row = threadIdx.y + (blockDim.y * blockIdx.y);

  if ((row < h) && (col < w)){
    float *myrow = (float *)(((char *)map)+(row*pitch));
    myrow[col] = 0.5f;
}

这假设pitch参数将作为字节数传递，这是cudaMallocPitch设置它的方式。所以我们要做的第一件事就是转换map为char指针。然后我们根据选定的行为其添加适当的音高偏移，并将结果指针转换为float指针。然后我们使用指针索引到选定的行float。

因此，您的内核调用应调整为：

initMap <<<numBlocks, numThr>>> (map, pitch, width, height, 1, 1);

请注意，索引大于w行的任何“幻像”元素都不会由此初始化。它们将包含垃圾。但是您不应该担心这一点或使用这些元素。

当您将此数组复制回主机时，您应该使用cudaMemcpy2D，如果设置正确，它将在复制回主机的过程中消除幻像元素，因此主机数组中应该没有垃圾数据。SO上有很多问题展示了如何正确使用cudaMallocPitch/cudaMemcpy2D。

于 2014-02-15T18:23:40.733 回答