require targets to provide whether they are executable

Author: kaushikcfd

loopy/target/__init__.py

@@ -49,6 +49,7 @@
 
 from typing import (Any, Tuple, Generic, TypeVar, Sequence, ClassVar, Optional,
         TYPE_CHECKING)
+import abc
 
 if TYPE_CHECKING:
     from loopy.typing import ExpressionT
@@ -159,6 +160,13 @@ def get_kernel_executor(self, kernel, *args, **kwargs):
         """
         raise NotImplementedError()
 
+    @abc.abstractproperty
+    def is_executable(self) -> bool:
+        """
+        Returns *True* only if the target allows executing loopy
+        translation units through :attr:`loopy.TranslationUnit.__call__`.
+        """
+
 
 class ASTBuilderBase(Generic[ASTType]):
     """An interface for generating (host or device) ASTs.

loopy/target/c/__init__.py

@@ -859,7 +859,9 @@ def get_function_declaration(self, codegen_state: CodeGenerationState,
             # subkernel launches occur only as part of entrypoint kernels for now
             from loopy.schedule.tools import get_subkernel_arg_info
             skai = get_subkernel_arg_info(kernel, subkernel_name)
-            passed_names = skai.passed_names
+            passed_names = (skai.passed_names
+                            if self.target.is_executable
+                            else [arg.name for arg in kernel.args])
             written_names = skai.written_names
         else:
             name = Value("static void", name)
@@ -1352,6 +1354,10 @@ def get_dtype_registry(self):
         fill_registry_with_c99_complex_types(result)
         return DTypeRegistryWrapper(result)
 
+    @property
+    def is_executable(self) -> bool:
+        return False
+
 
 class CASTBuilder(CFamilyASTBuilder):
     def preamble_generators(self):
@@ -1388,6 +1394,10 @@ def get_host_ast_builder(self):
         # enable host code generation
         return CFamilyASTBuilder(self)
 
+    @property
+    def is_executable(self) -> bool:
+        return True
+
 # }}}
 
 

loopy/target/cuda.py

@@ -264,6 +264,10 @@ def vector_dtype(self, base, count):
 
     # }}}
 
+    @property
+    def is_executable(self) -> bool:
+        return False
+
 # }}}
 
 

loopy/target/ispc.py

@@ -198,6 +198,10 @@ def get_dtype_registry(self):
 
     # }}}
 
+    @property
+    def is_executable(self) -> bool:
+        return False
+
 
 class ISPCASTBuilder(CFamilyASTBuilder):
     # {{{ top-level codegen
@@ -220,7 +224,9 @@ def get_function_declaration(self, codegen_state: CodeGenerationState,
             # subkernel launches occur only as part of entrypoint kernels for now
             from loopy.schedule.tools import get_subkernel_arg_info
             skai = get_subkernel_arg_info(codegen_state.kernel, subkernel_name)
-            passed_names = skai.passed_names
+            passed_names = (skai.passed_names
+                            if self.target.is_executable
+                            else [arg.name for arg in kernel.args])
             written_names = skai.written_names
         else:
             passed_names = [arg.name for arg in kernel.args]
@@ -261,7 +267,7 @@ def get_kernel_call(self, codegen_state: CodeGenerationState,
                         "assert(programCount == (%s))"
                         % ecm(lsize[0], PREC_NONE)))
 
-        if codegen_state.is_entrypoint:
+        if codegen_state.is_entrypoint and self.target.is_executable:
             # subkernel launches occur only as part of entrypoint kernels for now
             from loopy.schedule.tools import get_subkernel_arg_info
             skai = get_subkernel_arg_info(codegen_state.kernel, subkernel_name)

loopy/target/opencl.py

@@ -594,6 +594,10 @@ def is_vector_dtype(self, dtype):
     def vector_dtype(self, base, count):
         return NumpyType(vec.types[base.numpy_dtype, count])
 
+    @property
+    def is_executable(self) -> bool:
+        return False
+
 # }}}
 
 

loopy/target/pyopencl.py

@@ -526,6 +526,10 @@ def with_device(self, device):
                 "stop working in 2022.", DeprecationWarning, stacklevel=2)
         return self
 
+    @property
+    def is_executable(self) -> bool:
+        return True
+
 # }}}