Introduce PyOpenCLTarget.with_device to maintain target type

loopy/target/pyopencl.py

@@ -411,6 +411,9 @@ class PyOpenCLTarget(OpenCLTarget):
         from loopy.target.pyopencl_execution import PyOpenCLKernelExecutor
         return PyOpenCLKernelExecutor(queue.context, kernel)
 
+    def with_device(self, device):
+        return type(self)(device)
+
 # }}}
 
 

loopy/target/pyopencl_execution.py

@@ -267,7 +267,8 @@ class PyOpenCLKernelExecutor(KernelExecutorBase):
 
         from loopy.target.pyopencl import PyOpenCLTarget
         if isinstance(kernel.target, PyOpenCLTarget):
-            self.kernel = kernel.copy(target=PyOpenCLTarget(context.devices[0]))
+            self.kernel = kernel.copy(target=(
+                kernel.target.with_device(context.devices[0])))
 
     def get_invoker_uncached(self, kernel, codegen_result):
         generator = PyOpenCLExecutionWrapperGenerator()