deduplicate getConstPropPatterns() implementation

Signed-off-by: Soren Lassen <[email protected]>
onnx · Sep 12, 2023 · 2524321 · 2524321
1 parent 3006263
commit 2524321
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/src/Transform/ONNX/ConstProp.cpp b/src/Transform/ONNX/ConstProp.cpp
@@ -1013,12 +1013,6 @@ class SplitOfConst : public OpRewritePattern<ONNXSplitOp> {
   }
 };
 
-void getPatterns(RewritePatternSet &patterns) {
-  populateWithGenerated(patterns);
-  if (isNotDisabled("SplitOfConst"))
-    patterns.insert<SplitOfConst>(patterns.getContext());
-}
-
 //===----------------------------------------------------------------------===//
 // Code to manage the pass.
 //===----------------------------------------------------------------------===//
@@ -1042,15 +1036,17 @@ void ConstPropONNXToONNXPass::runOnOperation() {
   MLIRContext *context = &getContext();
 
   RewritePatternSet patterns(context);
-  getPatterns(patterns);
+  getConstPropPatterns(patterns);
   if (failed(applyPatternsAndFoldGreedily(function, std::move(patterns))))
     signalPassFailure();
 }
 
 } // end anonymous namespace.
 
 void onnx_mlir::getConstPropPatterns(RewritePatternSet &patterns) {
-  getPatterns(patterns);
+  populateWithGenerated(patterns);
+  if (isNotDisabled("SplitOfConst"))
+    patterns.insert<SplitOfConst>(patterns.getContext());
 }
 
 void onnx_mlir::configureConstPropONNXToONNXPass(