Remove support for kokkos main tranlation

Only used by workloads

Author: kennykos

pykokkos/core/translators/bindings.py

@@ -4,7 +4,7 @@
 
 from pykokkos.core import cppast
 from pykokkos.core.keywords import Keywords
-from pykokkos.core.visitors import cpp_view_type, KokkosMainVisitor, visitors_util
+from pykokkos.core.visitors import cpp_view_type, visitors_util
 from pykokkos.interface.data_types import DataType
 
 from .members import PyKokkosMembers
@@ -564,49 +564,6 @@ def bind_workunits(
     return bindings
 
 
-def translate_mains(
-    source: Tuple[List[str], int],
-    functor: str,
-    members: PyKokkosMembers,
-    pk_import: str,
-) -> List[str]:
-    """
-    Translate all PyKokkos main functions
-
-    :param source: the python source code of the workload
-    :param functor: the name of the functor
-    :param members: an object containing the fields and views
-    :returns: a list of strings of translated source code
-    """
-
-    node_visitor = KokkosMainVisitor(
-        {},
-        source,
-        members.views,
-        members.pk_workunits,
-        members.fields,
-        members.pk_functions,
-        members.classtype_methods,
-        functor,
-        pk_import,
-        debug=True,
-    )
-
-    translation: List[str] = []
-
-    for main in members.pk_mains.values():
-        try:
-            translation.append(node_visitor.visit(main))
-        except NotImplementedError:
-            print(f"Translation of {main.name} failed")
-            sys.exit(1)
-
-    members.reduction_result_queue = node_visitor.reduction_result_queue
-    members.timer_result_queue = node_visitor.timer_result_queue
-
-    return translation
-
-
 def bind_main_single(
     functor: str,
     members: PyKokkosMembers,

pykokkos/core/translators/members.py

@@ -8,7 +8,6 @@
 from pykokkos.core.parsers import PyKokkosEntity, PyKokkosStyles
 from pykokkos.core.visitors import (
     ConstructorVisitor,
-    KokkosMainVisitor,
     ParameterVisitor,
     visitors_util,
 )
@@ -154,45 +153,6 @@ def get_views(
 
         return views
 
-    def get_queues(
-        self, source: Tuple[List[str], int], name: str, pk_import: str
-    ) -> Tuple[List[str], List[str]]:
-        """
-        Get all fields assigned to a reduction result or timer result
-
-        :param source: the python source code of the workload
-        :param name: the name of the workload
-        :param pk_import: the identifier used to access the PyKokkos package
-        :returns: two lists, one for the reduction results and one for the timer results
-        """
-
-        views = copy.deepcopy(
-            self.views
-        )  # Needed since KokkosMainVisitor modifies views
-
-        # Copied from translate_mains() in bindings.py
-        node_visitor = KokkosMainVisitor(
-            {},
-            source,
-            views,
-            self.pk_workunits,
-            self.fields,
-            self.pk_functions,
-            self.classtype_methods,
-            name,
-            pk_import,
-            debug=True,
-        )
-
-        for main in self.pk_mains.values():
-            try:
-                node_visitor.visit(main)
-            except NotImplementedError:
-                print(f"Translation of {main.name} failed")
-                sys.exit(1)
-
-        return (node_visitor.reduction_result_queue, node_visitor.timer_result_queue)
-
     def get_real_views(self):
         """
         Get all the views that contain a pk.real datatype

pykokkos/core/visitors/__init__.py

@@ -2,7 +2,6 @@
 from .constructor_visitor import ConstructorVisitor
 from .debug_transformer import DebugTransformer
 from .kokkosfunction_visitor import KokkosFunctionVisitor
-from .kokkosmain_visitor import KokkosMainVisitor
 from .parameter_visitor import ParameterVisitor
 from .pykokkos_visitor import PyKokkosVisitor
 from .visitors_util import cpp_view_type, parse_view_template_params