misc/compiler: Rename/separate files, clean up and add lower_petsc_symbols

Author: ZoeLeibowitz

devito/core/cpu.py

@@ -14,6 +14,8 @@
                                check_stability, PetscTarget)
 from devito.tools import timed_pass
 
+from devito.petsc.iet.passes import lower_petsc_symbols
+
 __all__ = ['Cpu64NoopCOperator', 'Cpu64NoopOmpOperator', 'Cpu64AdvCOperator',
            'Cpu64AdvOmpOperator', 'Cpu64FsgCOperator', 'Cpu64FsgOmpOperator',
            'Cpu64CustomOperator', 'Cpu64CustomCXXOperator', 'Cpu64AdvCXXOperator',
@@ -143,6 +145,9 @@ def _specialize_iet(cls, graph, **kwargs):
         # Symbol definitions
         cls._Target.DataManager(**kwargs).process(graph)
 
+        # Lower PETSc symbols
+        lower_petsc_symbols(graph, **kwargs)
+
         return graph
 
 
@@ -222,6 +227,9 @@ def _specialize_iet(cls, graph, **kwargs):
         # Symbol definitions
         cls._Target.DataManager(**kwargs).process(graph)
 
+        # Lower PETSc symbols
+        lower_petsc_symbols(graph, **kwargs)
+
         # Linearize n-dimensional Indexeds
         linearize(graph, **kwargs)
 

devito/ir/iet/visitors.py

@@ -1068,7 +1068,7 @@ class FindSymbols(LazyVisitor[Any, list[Any], None]):
         Drive the search. Accepted:
         - `symbolics`: Collect all AbstractFunction objects, default
         - `basics`: Collect all Basic objects
-        - `abstractsymbols`: Collect all AbstractSymbol objects
+        - `symbols`: Collect all AbstractSymbol objects
         - `dimensions`: Collect all Dimensions
         - `indexeds`: Collect all Indexed objects
         - `indexedbases`: Collect all IndexedBase objects

devito/passes/iet/languages/C.py

@@ -7,6 +7,7 @@
 from devito.passes.iet.langbase import LangBB
 from devito.symbolics import c_complex, c_double_complex
 from devito.tools import dtype_to_cstr
+
 from devito.petsc.utils import petsc_type_mappings
 
 __all__ = ['CBB', 'CDataManager', 'COrchestrator']
@@ -82,3 +83,6 @@ class PetscCPrinter(CPrinter):
     _restrict_keyword = ''
 
     type_mappings = {**CPrinter.type_mappings, **petsc_type_mappings}
+
+    def _print_Pi(self, expr):
+        return 'PETSC_PI'