CUDA 内核 - 嵌套 for 循环

Question

你好 我正在尝试编写一个 CUDA 内核来执行以下代码。

for (n = 0; n < (total-1); n++)
{
  a = values[n];

  for ( i = n+1; i < total ; i++)
  {
    b = values[i] - a;
    c = b*b;

    if( c < 10)
        newvalues[i] = c;
    }
}

这是我目前拥有的，但似乎没有给出正确的结果？有谁知道我做错了什么。干杯

__global__ void calc(int total, float *values, float *newvalues){

float a,b,c;

int idx = blockIdx.x * blockDim.x + threadIdx.x;

for (int n = idx; n < (total-1); n += blockDim.x*gridDim.x){
    a = values[n];

    for(int i = n+1; i < total; i++){
        b = values[i] - a;
        c = b*b;

    if( c < 10)
        newvalues[i] = c;

    }
}

Answer 1

在 2D 中实现这个问题并使用 2D 线程块启动您的内核。 x 和 y 维度的线程总数将等于 total 。内核代码应如下所示：

__global__ void calc(float *values, float *newvalues, int total){


float a,b,c;

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

  if (n>=total || i>=total)
        return;

a = values[n];
b = values[i] - a;
c = b*b;
 if( c < 10)
        newvalues[i] = c;  

// I don't know your problem statement but i think it should be like: newvalues[n*total+i] = c;  


}

更新：

这就是你应该如何调用内核

dim3 block(16,16);
dim3 grid (  (total+15)/16,  (total+15)/16  );
calc<<<grid,block>>>(float *val, float *newval, int T);

还要确保在内核中添加这一行（请参阅更新的内核）

if (n>=total || i>=total)
return;

Answer 2

我可能错了，但 n < (total-1) 签到

for (int n = idx; n < (total-1); n += blockDim.x*gridDim.x)

似乎与原始版本不同。

Answer 3

您为什么不直接删除外部循环并使用该循环所需的尽可能多的线程启动内核呢？有一个取决于您的 blockId 的循环有点奇怪。通常你会尽量避免这些循环。 其次，在我看来newvalues[i] 可以被不同的线程覆盖。