CUDA 到 OpenCL：OpenCL 中 (blockIdx.x + blockIdx.y*gridDim.x) * blockDim.x + threadIdx.x 的等价物是什么？

Question

我是 OpenCL 的初学者，试图将一个简单的 CUDA 函数转换为 OpenCL。在 CUDA 函数中，他们使用下面的 sn-p 来获取操作索引，

int id = (blockIdx.x + blockIdx.y*gridDim.x) * blockDim.x + threadIdx.x;

get_global_id(0) 是否相当于 OpenCL 中的上述内容？

int id = get_global_id(0);

Answer 1

我不熟悉 CUDA（仅限 OpenCL），但看起来以下是等价的：

• get_group_id(uint dimindx) 和 blockIdx.[xyz]
• get_local_size(uint dimindx) 和 blockDim.[xyz]
• get_local_id(uint dimindx) 和 threadIdx.[xyz]
• get_num_groups(uint dimindx) 和 gridDim.[xyz]

其中dimindx 为0、1 或2，分别对应.x、.y 或.z。

get_global_id(0) 与get_group_id(0) * get_local_size(0) + get_local_id(0) 相同，因此假设上述等价确实正确，则与blockIdx.x * blockDim.x + threadIdx.x 相同。 （对于 .y 的索引 1 和 .z 的索引 2 等效）

要获得相同的 ID 值，我想你可能想要这样的东西：

int id = get_group_id(1) * get_global_size(0) + get_global_id(0);

请注意，原始 CUDA 代码明确不考虑 threadIdx.y。我怀疑这可能是因为您的 blockDim.y 为 1，或者因为您真的试图获取块中最顶部的项目（不是与当前线程对应的项目）。

更常见的情况可能是将当前工作项的索引放入包含二维元素数组的缓冲区中的数组中：

int id = get_global_id(1) * get_global_size(0) + get_global_id(0);

如果get_local_size(1) 为 1，这当然等价于前面的表达式。（块的高度为 1。）

以上所有内容均假设您已将内核与 CUDA 中的等效工作组（块）和全局维度排入队列。