本文共 401 字,大约阅读时间需要 1 分钟。
一、概述
在GPU编程上,通常我们可能会创建成千上万个并发线程来实现设备上的高吞吐量。例如:
这将会调用名为 some_kernel_func 的GPU函数共 2*128 次,每次都是不同的线程。然而,这样做通常会使 thread_idx 参数的计算变得更加复杂,而 thread_idx 通常又用来表示数组的下标。因此,内核函数如下:
为了计算thread_idx 这个参数,我们必须考虑线程块的数量。对第一个线程块而言,blockIdx.x 是0,因此 thread_idx 直接就等于之前使用过的 threadx.x,然而,对于第二个线程块而言,它的blockIdx.x 的值是1,blockDim.x 表示本例中所要求的每个线程块启动的线程数量,它的值是128,那么对第二个线程块而言,在计算thread_idx 时,要在threadIdx.x 的基础上加上一个1*128线程的基地址。
例如: