CUDA 中的顺序编程

Question

我想在 CUDA 中实现这个简单的循环。

for (int i = 1; i <= N; i++)
    {
        for (int j = 1; j <= N; j++)
        {
        x[i, j] = (x0[i, j] + a*(x[i - 1, j] + x[i + 1, j] + x[i, j - 1] + x[i, j + 1])) / c;
        }
    }

问题是：计算例如X[i,j] 我需要知道 X[i-1,j] 和 X[i,j-1] 的新值，如果我想在 CPU 上计算它很容易（计算是连续的）。但是 GPU 正在并行计算所有内容，因此从 CPU 和 GPU 收到的结果是不同的。我在 CUDA 和 cudaDeviceSynchronize() 中找到了一些关于动态并行的信息，并相信它可能很有用，但无论如何仍然不知道如何在内核中实现这个循环。我会很感激你的帮助。

Answer 1

上面的cmets是对的，即使是纯顺序实现，也需要复制数据。这里可能是你的内核（没有内存管理代码或任何进一步的细节）：

   __global__ void update(...)
    for(int i = threadIdx.x + blockDim.x * blockIdx.x; i <= N; i += blockDim.x * gridDim.x)
{
    for(int j = threadIdx.y + blockDim.y * blockIdx.y; j <= N; j += blockDim.y * gridDim.y)
    {
        output[i,j] = update_func(input, i, j);
    }
}

你可以调用（从主机），使用

update<<<dim3(16, 16), dim3(64, 64)>>>(input, output, width, height);

用适合你的硬件的任何值替换发射边界