예제 #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)