Fix analysis of rewrites where operation is inaccesible

xdslproject · Nov 28, 2023 · 068376a · 068376a
1 parent ddad126
commit 068376a
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/xdsl_pdl/analysis/pdl_analysis.py b/xdsl_pdl/analysis/pdl_analysis.py
@@ -492,17 +492,19 @@ def _analyze_rhs_op(self, rhs_op: Operation) -> AnalyzedPDLOperation | None:
             # Get the operation that is going to be replaced
             assert isinstance(rhs_op.op_value.owner, pdl.OperationOp)
             if (analyzed_op := self.get_analysis(rhs_op.op_value.owner)) is None:
-                raise PDLAnalysisException(
-                    rhs_op, "Unknown pdl.Operation to be replaced!"
+                raise PDLAnalysisAborted(
+                    rhs_op,
+                    "pdl.Operation to be replaced is not part of the matched DAG!",
                 )
             # Replacing with an operation
             if rhs_op.repl_operation:
                 assert isinstance(rhs_op.repl_operation.owner, pdl.OperationOp)
                 if (
                     analyzed_repl_op := self.get_analysis(rhs_op.repl_operation.owner)
                 ) is None:
-                    raise PDLAnalysisException(
-                        rhs_op, "Unknown pdl.Operation to be replaced!"
+                    raise PDLAnalysisAborted(
+                        rhs_op,
+                        "pdl.Operation used for replacement is not part of the matched DAG!",
                     )
                 analyzed_op.replaced_by = analyzed_repl_op