Rebase Refactor passes module & clean-up exports with v0.10.0-rc

frontend/catalyst/__init__.py

@@ -81,15 +81,15 @@
     "mlir_quantum._mlir_libs._quantumDialects.mitigation"
 )
 
-from catalyst import debug, logging
+from catalyst import debug, logging, passes
 from catalyst.api_extensions import *
 from catalyst.api_extensions import __all__ as _api_extension_list
 from catalyst.autograph import *
 from catalyst.autograph import __all__ as _autograph_functions
 from catalyst.compiler import CompileOptions
 from catalyst.debug.assertion import debug_assert
 from catalyst.jit import QJIT, qjit
-from catalyst.passes import Pass, PassPlugin, apply_pass, apply_pass_plugin, pipeline
+from catalyst.passes.pass_api import pipeline
 from catalyst.utils.exceptions import (
     AutoGraphError,
     CompileError,
@@ -187,11 +187,8 @@
     "debug_assert",
     "CompileOptions",
     "debug",
-    "apply_pass",
-    "apply_pass_plugin",
+    "passes",
     "pipeline",
-    "Pass",
-    "PassPlugin",
     *_api_extension_list,
     *_autograph_functions,
 )

frontend/catalyst/jax_primitives_utils.py

@@ -274,8 +274,9 @@ def transform_named_sequence_lowering(jax_ctx: mlir.LoweringRuleContext, pipelin
         with ir.InsertionPoint(bb_named_sequence):
             target = bb_named_sequence.arguments[0]
             for _pass in pipeline:
+                options = _pass.get_options()
                 apply_registered_pass_op = ApplyRegisteredPassOp(
-                    result=transform_mod_type, target=target, pass_name=_pass.name
+                    result=transform_mod_type, target=target, pass_name=_pass.name, options=options
                 )
                 target = apply_registered_pass_op.result
             transform_yield_op = YieldOp(operands_=[])  # pylint: disable=unused-variable