cuda - 线程块数之间的关系

Question

这段代码来自一本名为 cuda by examples

#include "../common/book.h"
#define N (33 * 1024)

__global__ void add( int *a, int *b, int *c ) {
    int tid = threadIdx.x + blockIdx.x * blockDim.x;
    while (tid < N) {
        c[tid] = a[tid] + b[tid];
        tid += blockDim.x * gridDim.x;
    }  
}

                   .
                   .
                   .
add<<<128,128>>>( dev_a, dev_b, dev_c );

33*1024 = 33792

128 * 128 = 16384

33792 > 16384

那么，在这种情况下，我是否必须增加每个块的线程数才能运行？

Answer 1

注意 while-cycle 主体中的第二个命令，即tid += blockDim.x * gridDim.x;. 它甚至可以处理比 16384 更大的数组。

ID 为 0 的线程将位置 0、16384、32768、...
中的数组项求和