Remove workload functionality

Author: kennykos

pykokkos/core/module_setup.py

@@ -52,22 +52,15 @@ def get_metadata(entity: Union[Callable[..., None], object]) -> EntityMetadata:
     name: str
     filepath: str
 
-    if isinstance(entity, Callable):
-        # Workunit/functor
-        is_functor: bool = hasattr(entity, "__self__")
-        if is_functor:
-            entity_type: type = get_functor(entity)
-            name = entity_type.__name__
-            filepath = inspect.getfile(entity_type)
-        else:
-            name = entity.__name__
-            filepath = inspect.getfile(entity)
-
-    else:
-        # Workload
-        entity_type: type = type(entity)
+    # Workunit/functor
+    is_functor: bool = hasattr(entity, "__self__")
+    if is_functor:
+        entity_type: type = get_functor(entity)
         name = entity_type.__name__
         filepath = inspect.getfile(entity_type)
+    else:
+        name = entity.__name__
+        filepath = inspect.getfile(entity)
 
     return EntityMetadata(entity, name, filepath)
 

pykokkos/core/parsers/parser.py

@@ -36,7 +36,7 @@ class PyKokkosEntity:
 
 class Parser:
     """
-    Parse a PyKokkos workload and its dependencies
+    Parse a PyKokkos workunit and its dependencies
     """
 
     def __init__(self, path: Optional[str], pk_import: Optional[str] = None):
@@ -55,7 +55,6 @@ def __init__(self, path: Optional[str], pk_import: Optional[str] = None):
                 self.tree = ast.parse("".join(self.lines))
             self.path: Optional[str] = path
             self.pk_import: str = self.get_import()
-            self.workloads: Dict[str, PyKokkosEntity] = {}
             self.classtypes: Dict[str, PyKokkosEntity] = {}
             self.functors: Dict[str, PyKokkosEntity] = {}
             self.workunits: Dict[str, PyKokkosEntity] = {}
@@ -109,8 +108,6 @@ def get_entity(self, name: str) -> PyKokkosEntity:
         :returns: the PyKokkosEntity representation of the entity
         """
 
-        if name in self.workloads:
-            return self.workloads[name]
         if name in self.functors:
             return self.functors[name]
         if name in self.workunits:

pykokkos/core/translators/members.py

@@ -120,7 +120,7 @@ def get_fields(
         :returns: a dictionary mapping from field name to type
 
 
-        NOTE: **used by workloads & functors; depreciate with functions** 
+        NOTE: **used by workloads & functors; depreciate with functions**
         """
 
         visitor = ConstructorVisitor(source, "fields", pk_import, True)
@@ -140,7 +140,7 @@ def get_views(
         :param pk_import: the identifier used to access the PyKokkos package
         :returns: a dictionary mapping from view name to type (only dimensionality and type)
 
-        NOTE: **used by workloads & functors; depreciate with functions** 
+        NOTE: **used by workloads & functors; depreciate with functions**
         """
 
         visitor = ConstructorVisitor(source, "views", pk_import, True)
@@ -277,7 +277,7 @@ def get_random_pool(
         :param pk_import: the identifier used to access the PyKokkos package
         :returns: the type of the random pool if it exists
 
-        NOTE: **used by workloads & functors; depreciate with functions** 
+        NOTE: **used by workloads & functors; depreciate with functions**
         """
 
         visitor = ConstructorVisitor(source, "randpool", pk_import, True)

pykokkos/core/translators/static.py

@@ -34,7 +34,7 @@ def generate_include_guard_end() -> str:
 
 class StaticTranslator:
     """
-    Translates a PyKokkos workload to C++ using static analysis only
+    Translates a PyKokkos workunit to C++ using static analysis only
     """
 
     def __init__(
@@ -189,9 +189,9 @@ def translate_classtypes(
         self, classtypes: List[PyKokkosEntity], restrict_views: Set[str]
     ) -> List[cppast.RecordDecl]:
         """
-        Translate all classtypes, i.e. classes that the workload uses internally
+        Translate all classtypes, i.e. classes that the workunit uses internally
 
-        :param classtypes: the list of classtypes needed by the workload
+        :param classtypes: the list of classtypes needed by the workunit
         :param restrict_views: the views with the restrict keyword
         :returns: a list of strings of translated source code
         """
@@ -363,7 +363,7 @@ def translate_functions(
         """
         Translate all PyKokkos functions
 
-        :param source: the python source code of the workload
+        :param source: the python source code of the workunit
         :param restrict_views: the views with the restrict keyword
         :returns: a list of method declarations
         """
@@ -400,9 +400,9 @@ def translate_workunits(
         """
         Translate the workunits
 
-        :param source: the python source code of the workload
+        :param source: the python source code of the workunit
         :param restrict_views: the views with the restrict keyword
-        :returns: a tuple of a dictionary mapping from workload name
+        :returns: a tuple of a dictionary mapping from workunit name
             to a tuple of operation name and source, and a boolean
             indicating whether the workunit has a call to pk.rand()
         """

pykokkos/core/visitors/__init__.py

@@ -1,6 +1,5 @@
 from .classtype_visitor import ClasstypeVisitor
 from .constructor_visitor import ConstructorVisitor
-from .debug_transformer import DebugTransformer
 from .kokkosfunction_visitor import KokkosFunctionVisitor
 from .parameter_visitor import ParameterVisitor
 from .pykokkos_visitor import PyKokkosVisitor

pykokkos/core/visitors/constructor_visitor.py

@@ -23,7 +23,7 @@ def __init__(
         :param pk_import: the identifier used to access the PyKokkos package
         :param debug: if true, prints the python AST when an error is encountered
 
-        NOTE: **used by workloads & functors; depreciate with functions** 
+        NOTE: **used by workloads & functors; depreciate with functions**
         """
 
         if member_type not in ("fields", "views", "typeinfo", "randpool"):

pykokkos/core/visitors/parameter_visitor.py

@@ -17,7 +17,7 @@ def __init__(
         """
         ParameterVisitor constructor
 
-        :param src: the python source code of the workload
+        :param src: the python source code of the workunit
         :param param_begin: where workunit argument begins (excluding tid/acc)
         :param pk_import: the identifier used to access the PyKokkos package
         :param debug: if true, prints the python AST when an error is encountered