OpenCL 使用工作组大小

Question

在 OpenCL 中，我如何知道运行时本地工作组的确切大小？ clGetKernelWorkGroupInfo 似乎只返回最大可能的大小，请参阅https://www.khronos.org/registry/cl/sdk/1.1/docs/man/xhtml/clGetKernelWorkGroupInfo.html

如果您没有指定大小，则上述文档的CL_KERNEL_COMPILE_WORK_GROUP_SIZE 将给出(0,0,0)。

Answer 1

您可以使用 clGetKernelWorkGroupInfo 查询最大可用工作组大小：

size_t max_available_local_wg_size;
cl_int ret = clGetKernelWorkGroupInfo(kernel, device_id, CL_KERNEL_WORK_GROUP_SIZE, sizeof(size_t),                 &max_available_local_wg_size, NULL);

但是，如果您不手动设置值，实现将自行选择它。通常，您可以使用供应商提供的 OpenCL SDK 随附的 profiler 获得实际的工作组大小。