Rewrite MLPatternLoweringPass to no longer inherit from FunctionPass and just provide a utility function that applies ML patterns.

PiperOrigin-RevId: 235194034

1 files changed, 8 insertions, 5 deletions

diff --git a/mlir/lib/Transforms/LowerVectorTransfers.cpp b/mlir/lib/Transforms/LowerVectorTransfers.cpp
index ac8f7e064f5..61f75ae76e6 100644
--- a/mlir/lib/Transforms/LowerVectorTransfers.cpp
+++ b/mlir/lib/Transforms/LowerVectorTransfers.cpp
@@ -424,12 +424,15 @@ public:
   }
 };
 
-struct LowerVectorTransfersPass
-    : public MLPatternLoweringPass<
-          VectorTransferExpander<VectorTransferReadOp>,
-          VectorTransferExpander<VectorTransferWriteOp>> {
+struct LowerVectorTransfersPass : public FunctionPass {
   LowerVectorTransfersPass()
-      : MLPatternLoweringPass(&LowerVectorTransfersPass::passID) {}
+      : FunctionPass(&LowerVectorTransfersPass::passID) {}
+
+  PassResult runOnFunction(Function *fn) override {
+    applyMLPatternsGreedily<VectorTransferExpander<VectorTransferReadOp>,
+                            VectorTransferExpander<VectorTransferWriteOp>>(fn);
+    return success();
+  }
 
   // Thread-safe RAII context with local scope. BumpPtrAllocator freed on exit.
   edsc::ScopedEDSCContext raiiContext;