CUDAネストループ

Question

CコードをCUDAに移植したいと思います。主な計算部分にはネストされたループのための3が含まれています：

for (int i=0; i< Nx;i++){ 
    for (int j=0;j<Ncontains[i];j++){ 
    for (int k=0;k< totalVoxels;k++){ 
      ....... 
    } 
    } 
} 

私はどのように私のCUDAカーネルに変換できますか？ 2つのforループを使用すると、次のようなことができます。

int n= blockIdy.y * blockDim.y + threadIdx.y; 
int i= blockIdx.x * blockDim.x + threadIdx.x; 

これはどうやって最初に起動できますか？あなたがそれを行うことができます

Answer 1

多くの方法、そのうちの一つは次のとおりです。

for (int i=blockIdx.x; i< Nx; i += gridDim.x){ 
    for (int j=threadIdx.y; j<Ncontains[i]; j+= blockDim.y){ 
    for (int k=threadIdx.x; k< totalVoxels; k += blockDim.x){ 
      ....... 
    } 
    } 
} 

あなたが呼ぶだろう上：

// nx,ny block dimensions 
kernel <<< dim3(nBlocks), dim3(nx, ny) >>> (...);