Update generic matcher for new parameter

microsoft · Dec 28, 2024 · 2b5309a · 2b5309a
1 parent 766791d
commit 2b5309a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/onnxscript/rewriter/generic_pattern.py b/onnxscript/rewriter/generic_pattern.py
@@ -551,7 +551,12 @@ def match(
         graph_or_function: ir.Graph | ir.Function,
         node: ir.Node,
         verbose: int = 0,
+        remove_nodes: bool = True,
     ) -> orp.MatchResult | None:
+        if not remove_nodes:
+            raise NotImplementedError(
+                "remove_nodes=False is not implemented in GenericPatternMatcher"
+            )
         del model
         del graph_or_function
         self.verbose = verbose