compiler: Callable.attributes

Author: FabioLuporini

devito/ir/iet/efunc.py

@@ -159,9 +159,9 @@ class DeviceFunction(Callable):
     """
 
     def __init__(self, name, body, retval='void', parameters=None,
-                 prefix='__global__', templates=None):
+                 prefix='__global__', templates=None, attributes=None):
         super().__init__(name, body, retval, parameters=parameters, prefix=prefix,
-                         templates=templates)
+                         templates=templates, attributes=attributes)
 
 
 class DeviceCall(Call):

devito/ir/iet/nodes.py

@@ -718,17 +718,21 @@ class Callable(Node):
     parameters : list of Basic, optional
         The objects in input to the Callable.
     prefix : list of str, optional
-        Qualifiers to prepend to the Callable signature. None by defaults.
+        Qualifiers to prepend to the Callable signature. None by default.
     templates : list of Basic, optional
         The template parameters of the Callable.
+    attributes : list of str, optional
+        Additional attributes to append to the Callable signature. An
+        attributes is one or more keywords that appear in between the
+        return type and the function name. None by default.
     """
 
     is_Callable = True
 
     _traversable = ['body']
 
     def __init__(self, name, body, retval, parameters=None, prefix=None,
-                 templates=None):
+                 templates=None, attributes=None):
         self.name = name
         if not isinstance(body, CallableBody):
             self.body = CallableBody(body)
@@ -738,6 +742,7 @@ def __init__(self, name, body, retval, parameters=None, prefix=None,
         self.prefix = as_tuple(prefix)
         self.parameters = as_tuple(parameters)
         self.templates = as_tuple(templates)
+        self.attributes = as_tuple(attributes)
 
     def __repr__(self):
         param_types = [ctypes_to_cstr(i._C_ctype) for i in self.parameters]

devito/ir/iet/visitors.py

@@ -320,14 +320,25 @@ def _args_call(self, args):
 
     def _gen_signature(self, o, is_declaration=False):
         decls = self._args_decl(o.parameters)
+
         prefix = ' '.join(o.prefix + (self._gen_rettype(o.retval),))
-        signature = c.FunctionDeclaration(c.Value(prefix, o.name), decls)
+
+        if o.attributes:
+            # NOTE: ugly, but I can't bother extending `c.FunctionDeclaration`
+            # for such a tiny thing
+            v = f"{' '.join(o.attributes)} {o.name}"
+        else:
+            v = o.name
+
+        signature = c.FunctionDeclaration(c.Value(prefix, v), decls)
+
         if o.templates:
             tparams = ', '.join([i.inline() for i in self._args_decl(o.templates)])
             if is_declaration:
                 signature = TemplateDecl(tparams, signature)
             else:
                 signature = c.Template(tparams, signature)
+
         return signature
 
     def _blankline_logic(self, children):