Cuda 三重嵌套 for 循环赋值

Question

我正在尝试将 C++ 代码转换为 Cuda 代码，并且我有以下三重嵌套 for 循环，它将填充一个数组以供进一步 OpenGL 渲染（我只是创建一个坐标顶点数组）：

for(int z=0;z<263;z++) {                    
       for(int y=0;y<170;y++) {
           for(int x=0;x<170;x++) {
               g_vertex_buffer_data_3[i]=(float)x+0.5f;
               g_vertex_buffer_data_3[i+1]=(float)y+0.5f;
               g_vertex_buffer_data_3[i+2]=-(float)z+0.5f; 
               i+=3;            
           }
       }
   }

我希望获得更快的操作，因此我将使用 Cuda 进行一些操作，例如上面列出的操作。我想为最外层循环的每次迭代创建一个块，并且由于内部循环的迭代次数为 170 * 170 = 28900 次，因此为每个最内层循环迭代分配一个线程。我将 c++ 代码转换成这个（这只是我为了解如何使用 Cuda 而编写的一个小程序）：

__global__ void mykernel(int k, float *buffer) {
int idz=blockIdx.x;
int idx=threadIdx.x;
int idy=threadIdx.y;

buffer[k]=idx+0.5;
buffer[k+1]=idy+0.5;
buffer[k+2]=idz+0.5;
k+=3;

}

int main(void) {
  int dim=3*170*170*263;
  float* g_vertex_buffer_data_2 = new float[dim];
  float* g_vertex_buffer_data_3;
  int i=0;

  HANDLE_ERROR(cudaMalloc((void**)&g_vertex_buffer_data_3, sizeof(float)*dim));

  dim3 dimBlock(170, 170);

  dim3 dimGrid(263);

  mykernel<<<dimGrid, dimBlock>>>(i, g_vertex_buffer_data_3);

  HANDLE_ERROR(cudaMemcpy(&g_vertex_buffer_data_2,g_vertex_buffer_data_3,sizeof(float)*dim,cudaMemcpyDeviceToHost));

  for(int j=0;j<100;j++){
    printf("g_vertex_buffer_data_2[%d]=%f\n",j,g_vertex_buffer_data_2[j]);
  }
  cudaFree(g_vertex_buffer_data_3);

  return 0;

}

尝试启动它时出现分段错误。你知道我做错了什么吗？我认为问题在于 threadIdx.x 和 threadIdx.y 同时增长，而我希望 threadIdx.x 成为内层而 threadIdx.y 成为外层.

Answer 1

这里有很多错误，但是段错误的来源是这样的：

cudaMemcpy(&g_vertex_buffer_data_2,g_vertex_buffer_data_3,
                sizeof(float)*dim,cudaMemcpyDeviceToHost);

你要么

cudaMemcpy(&g_vertex_buffer_data_2[0],g_vertex_buffer_data_3,
                sizeof(float)*dim,cudaMemcpyDeviceToHost);

或

cudaMemcpy(g_vertex_buffer_data_2,g_vertex_buffer_data_3,
                sizeof(float)*dim,cudaMemcpyDeviceToHost);

修复后，您会注意到内核实际上从未启动并出现无效启动错误。这是因为 (170,170) 的块大小是非法的。 CUDA 在所有当前硬件上每个块限制 1024 个线程。

您的代码中可能还有其他问题。找到这两个后我就不再找了。

Cuda 三重嵌套 for 循环赋值

Cuda triple nested for loop assignement

c++

for-loop

cuda

nested-loops