示例#1
0
    def get_kernel_executor(self, program, queue, **kwargs):
        from loopy.target.pyopencl_execution import PyOpenCLKernelExecutor

        epoint = kwargs.pop("entrypoint")
        program = self.preprocess_translation_unit_for_passed_args(
            program, epoint, kwargs)

        return PyOpenCLKernelExecutor(queue.context,
                                      program,
                                      entrypoint=epoint)
示例#2
0
文件: pyopencl.py 项目: shigh/loopy
 def get_kernel_executor(self, kernel, queue, **kwargs):
     from loopy.target.pyopencl_execution import PyOpenCLKernelExecutor
     return PyOpenCLKernelExecutor(queue.context, kernel)