Printing as a pass

Author: erick-xanadu

a.py

@@ -30,7 +30,7 @@ def identity_transform(tape):
 # All passes inherit from passes.ModulePass
 class PrintModule(passes.ModulePass):
     # All passes require a name field
-    name = "print"
+    name = "remove-chained-self-inverse"
 
     # All passes require an apply method with this signature.
     def apply(self, ctx: context.MLContext, module: builtin.ModuleOp) -> None:

frontend/catalyst/compiler.py

@@ -488,8 +488,9 @@ def run_python_compiler(self, mlir_module):
         # TODO: check that xdsl is available to be imported
         import xdsl
         from xdsl.context import Context
+        from xdsl import passes
         from xdsl.dialects import arith, builtin, func, scf, tensor, transform
-        from .python_compiler import quantum 
+        from .python_compiler import quantum, PrintModule
         generic_assembly_format = mlir_module.operation.get_asm(binary=False, print_generic_op_form=True, assume_verified=True)
         ctx = Context(allow_unregistered=True)
         ctx.load_dialect(arith.Arith)
@@ -507,6 +508,8 @@ def run_python_compiler(self, mlir_module):
         # TODO: Load Catalyst
         # TODO: Load ion/ppm/mbqc/zne...
         module = xdsl.parser.Parser(ctx, generic_assembly_format).parse_module()
+        pipeline = passes.PipelinePass((PrintModule(),))
+        pipeline.apply(ctx, module)
 
         from jax._src.interpreters import mlir
         from jaxlib.mlir.dialects import stablehlo
@@ -520,7 +523,6 @@ def run_python_compiler(self, mlir_module):
             ctx.allow_unregistered_dialects = True
             ctx.append_dialect_registry(mlir.upstream_dialects)
             stablehlo.register_dialect(ctx)
-
             module = Module.parse(buffer.getvalue())
         return module
 

frontend/catalyst/python_compiler/__init__.py

@@ -0,0 +1,16 @@
+from typing import Callable
+
+from dataclasses import dataclass
+from xdsl import context, passes
+from xdsl.utils import parse_pipeline
+from xdsl.dialects import builtin
+
+@dataclass(frozen=True)
+# All passes inherit from passes.ModulePass
+class PrintModule(passes.ModulePass):
+    # All passes require a name field
+    name = "print"
+
+    # All passes require an apply method with this signature.
+    def apply(self, ctx: context.MLContext, module: builtin.ModuleOp) -> None:
+        print("Hello from inside the pass\n", module)