Cuda索引圖3D塊，2D網格

Question

我是新來的cuda;我有一個2D圖像（寬度，高度）與3個通道（顏色）。 我想要的是午餐有3D塊和2D網格這樣

kernel_2D_3D<<<dim3(1,m,n), dim3(3,TILEy,TILEz)>>>(float *in, float *out) 

內核我用X的顏色，ÿ寬度和身高ž。我的問題是： 如何計算圖像的行和列：

1. unsigned int Row =？
2. unsigned int Col =？

和我用這個功能來計算全球唯一索引

__device__ int getGlobalIdx_2D_3D() 
{ 
    int blockId = blockIdx.x+ blockIdx.y * gridDim.x; 

    int Idx = blockId * (blockDim.x * blockDim.y * blockDim.z) 
       + (threadIdx.z * (blockDim.x * blockDim.y)) 
       + (threadIdx.y * blockDim.x) 
       + threadIdx.x; 

    return Idx; 
} 

Answer 1

如果您使用的Y表示寬度和z的高度，則圖像的行和列可以這樣來計算這裏面內核：

unsigned int row = blockIdx.z * blockDim.z + threadIdx.z; 
unsigned int col = blockIdx.y * blockDim.y + threadIdx.y; 

當前通道將等於threadIdx.x。