如何在 pyopencl 中明确设置工作维度？答案

【问题标题】：how to explicitely set work - dimension in pyopencl?如何在 pyopencl 中明确设置工作维度？
【发布时间】：2012-10-22 10:48:25
【问题描述】：

在用 C++ 编写 OpenCL 主机程序时，我们使用以下 API 来调用 OpenCL 内核：

cl_int clEnqueueNDRangeKernel (     cl_command_queue command_queue,
    cl_kernel kernel,
    cl_uint work_dim,
    const size_t *global_work_offset,
    const size_t *global_work_size,
    const size_t *local_work_size,
    cl_uint num_events_in_wait_list,
    const cl_event *event_wait_list,
    cl_event *event)

这里的第三个参数设置工作维度。在 python 中使用 pyopencl 时，我们将内核作为程序的一部分称为：

<program_name>.<kernel_name>( <command_queue>, <Global_work_size>,
                              <Local_work_size>, <Parameters_to_kernel.....> )

例如：

event = program.square( queue, A.shape, None, 
                        A_buf, B_buf, cl.LocalMemory( A.size), np.int32(COUNT) )

那么如何使用 pyopencl 在 python 中显式设置“work_dim”？

【问题讨论】：

标签： python opencl pyopencl

【解决方案1】：

你没有。 work_dim 在 C/C++ OpenCL API 中传递，以便库可以知道一维数组 global_work_offset 等的大小；在 PyOpenCL 中，这些大小是从作为 global_size、local_size 和 global_offset 传递的对象中推断出来的。

【讨论】：