commit c6c534493d625c10ce0046baa9dc6293f8dba405
author River Riddle <riverriddle@google.com> Wed Feb 27 18:59:29 2019
committer jpienaar <jpienaar@google.com> Fri Mar 29 23:47:14 2019
tree 87246fb531049bf478bb4b061217d11b8d7fe254
parent 6067cdebaa391cb2ab9bda9cd91b26f9ddd95a11

Port all of the existing passes over to the new pass manager infrastructure. This is largely NFC.

PiperOrigin-RevId: 235952357

mlir/lib/Transforms/LoopFusion.cpp

diff --git a/mlir/lib/Transforms/LoopFusion.cpp b/mlir/lib/Transforms/LoopFusion.cpp
index 0f4e45c..1528e39 100644
--- a/mlir/lib/Transforms/LoopFusion.cpp
+++ b/mlir/lib/Transforms/LoopFusion.cpp
@@ -86,14 +86,12 @@
 // TODO(andydavis) Extend this pass to check for fusion preventing dependences,
 // and add support for more general loop fusion algorithms.
 
-struct LoopFusion : public FunctionPass {
+struct LoopFusion : public FunctionPass<LoopFusion> {
   LoopFusion(unsigned fastMemorySpace = 0, uint64_t localBufSizeThreshold = 0)
-      : FunctionPass(&LoopFusion::passID),
-        localBufSizeThreshold(localBufSizeThreshold),
+      : localBufSizeThreshold(localBufSizeThreshold),
         fastMemorySpace(fastMemorySpace) {}
 
-  PassResult runOnFunction(Function *f) override;
-  constexpr static PassID passID = {};
+  PassResult runOnFunction() override;
 
   // Any local buffers smaller than this size (in bytes) will be created in
   // `fastMemorySpace` if provided.
@@ -107,8 +105,8 @@
 
 } // end anonymous namespace
 
-FunctionPass *mlir::createLoopFusionPass(unsigned fastMemorySpace,
-                                         uint64_t localBufSizeThreshold) {
+FunctionPassBase *mlir::createLoopFusionPass(unsigned fastMemorySpace,
+                                             uint64_t localBufSizeThreshold) {
   return new LoopFusion(fastMemorySpace, localBufSizeThreshold);
 }
 
@@ -1802,7 +1800,7 @@
 
 } // end anonymous namespace
 
-PassResult LoopFusion::runOnFunction(Function *f) {
+PassResult LoopFusion::runOnFunction() {
   // Override if a command line argument was provided.
   if (clFusionFastMemorySpace.getNumOccurrences() > 0) {
     fastMemorySpace = clFusionFastMemorySpace.getValue();
@@ -1814,7 +1812,7 @@
   }
 
   MemRefDependenceGraph g;
-  if (g.init(f))
+  if (g.init(&getFunction()))
     GreedyFusion(&g).run(localBufSizeThreshold, fastMemorySpace);
   return success();
 }