compiler: handle plain text header

Author: mloubout

devito/ir/iet/visitors.py

@@ -715,7 +715,15 @@ def visit_Operator(self, o, mode='all'):
             efuncs.extend([self._visit(i), blankline])
 
         # Definitions
-        headers = [c.Define(*i) for i in o._headers] + [blankline]
+        headers = []
+        for h in o._headers:
+            try:
+                headers.append(c.Define(*h))
+            except TypeError:
+                # Plain string
+                headers.append(c.Line(h))
+        headers = headers + [blankline]
+        # headers = [c.Define(*i) for i in o._headers] + [blankline]
 
         # Header files
         includes = self._operator_includes(o) + [blankline]

devito/passes/iet/dtypes.py

@@ -30,7 +30,7 @@ def _complex_includes(iet: Callable, lang: type[LangBB], compiler: Compiler,
         return iet, {}
 
     metadata = {}
-    lib = as_tuple(lang['header-complex'])
+    lib = as_tuple(lang['includes-complex'])
 
     if lang.get('complex-namespace') is not None:
         metadata['namespaces'] = lang['complex-namespace']

devito/passes/iet/languages/C.py

@@ -25,7 +25,7 @@ class CBB(LangBB):
         'alloc-global-symbol': lambda i, j, k:
             Call('memcpy', (i, j, k)),
         # Complex and float16
-        'header-complex': 'complex.h',
+        'includes-complex': 'complex.h',
     }
 
 

devito/passes/iet/languages/CXX.py

@@ -61,7 +61,7 @@ class CXXBB(LangBB):
         'alloc-global-symbol': lambda i, j, k:
             Call('memcpy', (i, j, k)),
         # Complex and float16
-        'header-complex': 'complex',
+        'includes-complex': 'complex',
         'complex-namespace': [UsingNamespace('std::complex_literals')],
         'def-complex': std_arith,
     }