嵌套 for 循环中的 Cuda 减少

Question

我有一个关于 CUDA 减少的问题。

distance 是具有gridSize*numberOfAngles 元素的矩阵，fftData 是具有numberOfAngles*NFFT 元素的矩阵。 grid_magnitude 是我要存储计算结果的结果矩阵，它有 gridSize 元素。

我想计算fftData 中对应于distance 中一个特定值的索引。之后，fftData 中此索引处的值将添加到对应的gridPoint 处的grid_magnitude。

这是我目前的代码：

__global__ void calcBackProjection(cuFloatComplex* fftData, 
                                   cuFloatComplex* grid_magnitude,
                                   float* distance,
                                   int gridSize,
                                   int numberOfAngles,
                                   float rangeBin,
                                   int NFFT)
{
 int gridPointIdx = threadIdx.x + blockIdx.x*blockDim.x;

 while(gridPointIdx < gridSize)
 {
    for(int angleIdx = 0; angleIdx < numberOfAngles; angleIdx++)
    {       
        //find rangeBin in fftData corresponding to distance
        float curDistance = distance[gridPointIdx + angleIdx*gridSize];
        int wantedIdx = floor(curDistance / rangeBin);
        if(wantedIdx < NFFT)
            {                                   
                grid_magnitude[gridPointIdx + angleIdx*gridSize] =  
              addCmplx(fftData[wantedIdx + angleIdx*NFFT], grid_magnitude[gridPointIdx +     
                angleIdx*gridSize]);

            }                   
    }
    gridPointIdx += blockDim.x * gridDim.x;     
 }   
}

gridPointIdx 对于每个线程应该是唯一的，因此每个线程应该在grid_magnitude 的不同位置写入。但似乎这不起作用，因为grid_magnitude 没有任何更改。

我错过了什么？

我没有设法在完全并行的二维索引中做到这一点，也许我应该使用共享内存，但我如何将 grid_magnitude 部分分配给线程使用？

Answer 1

我稍微更改了代码。

__global__ void calcBackProjection(cuFloatComplex* fftData, cuFloatComplex* grid_magnitude,
float* distance, int gridSize, int numberOfAngles, float rangeBin, int NFFT){
int gridPointIdx = threadIdx.x + blockIdx.x*blockDim.x;
while(gridPointIdx < gridSize){
    for(int angleIdx = 0; angleIdx < numberOfAngles; angleIdx++){       
        float curDistance = distance[gridPointIdx + angleIdx*gridSize];
        int wantedIdx = ceil(curDistance / rangeBin) - 1;
        if(wantedIdx){
            int fftIdx = wantedIdx + angleIdx*NFFT;
            int gridIdx=  gridPointIdx + angleIdx*gridSize;
            if((fftIdx < NFFT*numberOfAngles) && (gridIdx < gridSize*numberOfAngles)){                  
                grid_magnitude[gridIdx] = cuCaddf(fftData[fftIdx], grid_magnitude[gridIdx]);
            }
        }
    }
    gridPointIdx += blockDim.x * gridDim.x;     }}

问题似乎是，变量 curDistance 和 WantIdx 没有被编译器评估。当我想知道这些值时，它会说“在目标位置没有价值”。这似乎是在grid_magnitude[gridIdx] = cuCaddf(fftData[fftIdx], grid_magnitude[gridIdx]); 检测到访问冲突的原因 我查看了有关此问题的其他一些答案，例如 here 和 here，但这些对我帮助不大。