Remove main and callback decorators

Author: kennykos

pykokkos/core/translators/members.py

@@ -30,8 +30,6 @@ def __init__(self):
 
         self.pk_workunits: Dict[cppast.DeclRefExpr, ast.FunctionDef] = {}
         self.pk_functions: Dict[cppast.DeclRefExpr, ast.FunctionDef] = {}
-        self.pk_mains: Dict[cppast.DeclRefExpr, ast.FunctionDef] = {}
-        self.pk_callbacks: Dict[cppast.DeclRefExpr, ast.FunctionDef] = {}
 
         self.classtype_methods: Dict[cppast.DeclRefExpr, List[cppast.DeclRefExpr]] = {}
 
@@ -102,9 +100,6 @@ def extract(self, entity: PyKokkosEntity, classtypes: List[PyKokkosEntity]) -> N
             self.pk_functions = self.get_decorated_functions(
                 AST, Decorator.KokkosFunction
             )
-            self.pk_callbacks = self.get_decorated_functions(
-                AST, Decorator.KokkosCallback
-            )
         else:
             self.pk_workunits[cppast.DeclRefExpr(AST.name)] = AST
             self.pk_functions = self.get_decorated_functions(
@@ -113,10 +108,6 @@ def extract(self, entity: PyKokkosEntity, classtypes: List[PyKokkosEntity]) -> N
 
         self.classtype_methods = self.get_classtype_methods(classtypes)
 
-        if len(self.pk_mains) > 1:
-            print("ERROR: Only one pk.main function can be translated")
-            sys.exit(1)
-
     def get_fields(
         self, classdef: ast.ClassDef, source: Tuple[List[str], int], pk_import: str
     ) -> Dict[cppast.DeclRefExpr, cppast.PrimitiveType]:

pykokkos/core/translators/static.py

@@ -172,8 +172,6 @@ def check_symbols(self, classtypes: List[PyKokkosEntity], path: str) -> None:
         symbols_pass = SymbolsPass(self.pk_members, self.pk_import, path)
 
         error_messages: List[str] = []
-        for AST in self.pk_members.pk_mains.values():
-            error_messages.extend(symbols_pass.check_symbols(AST))
         for AST in self.pk_members.pk_workunits.values():
             error_messages.extend(symbols_pass.check_symbols(AST))
         for AST in self.pk_members.pk_functions.values():

pykokkos/interface/__init__.py

@@ -38,12 +38,10 @@
     complex128,
 )
 from .decorators import (
-    callback,
     classtype,
     Decorator,
     function,
     functor,
-    main,
     workunit,
 )
 from .execution_policy import (

pykokkos/interface/decorators.py

@@ -7,8 +7,6 @@ class Decorator(Enum):
     WorkUnit = "workunit"
     KokkosClasstype = "classtype"
     KokkosFunction = "function"
-    KokkosMain = "main"
-    KokkosCallback = "callback"
     Space = "space"
 
     @staticmethod
@@ -27,14 +25,6 @@ def is_kokkos_classtype(decorator: str) -> bool:
     def is_kokkos_function(decorator: str) -> bool:
         return decorator == Decorator.KokkosFunction.value
 
-    @staticmethod
-    def is_kokkos_main(decorator: str) -> bool:
-        return decorator == Decorator.KokkosMain.value
-
-    @staticmethod
-    def is_kokkos_callback(decorator: str) -> bool:
-        return decorator == Decorator.KokkosCallback.value
-
     @staticmethod
     def is_space(decorator: str) -> bool:
         return decorator == Decorator.Space.value
@@ -74,11 +64,3 @@ def classtype(func):
 
 def function(func):
     return func
-
-
-def main(func):
-    return func
-
-
-def callback(func):
-    return func