如何明确设置工作 - Pyopencl中的维度

Question

在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？

Answer 1

您不这样做。work_dim在C/C OpenCL API中传递，以便库可以知道一D数组global_work_offset的大小，等等；在pyopencl中，可以从传递的对象以global_size，local_size和global_offset。