1 files changed, 3 insertions, 4 deletions
diff --git a/mlir/lib/Transforms/ComposeAffineMaps.cpp b/mlir/lib/Transforms/ComposeAffineMaps.cpp
index 8f7919d0760..ced9cbcb86a 100644
--- a/mlir/lib/Transforms/ComposeAffineMaps.cpp
+++ b/mlir/lib/Transforms/ComposeAffineMaps.cpp
@@ -41,21 +41,20 @@ namespace {
 // also AffineApplyOps. After forward subtituting its results, AffineApplyOps
 // with no remaining uses are collected and erased after the walk.
 // TODO(andydavis) Remove this when Chris adds instruction combiner pass.
-struct ComposeAffineMaps : public MLFunctionPass,
-                           StmtWalker<ComposeAffineMaps> {
+struct ComposeAffineMaps : public FunctionPass, StmtWalker<ComposeAffineMaps> {
   std::vector<OperationStmt *> affineApplyOpsToErase;
 
   explicit ComposeAffineMaps() {}
   using StmtListType = llvm::iplist<Statement>;
   void walk(StmtListType::iterator Start, StmtListType::iterator End);
   void visitOperationStmt(OperationStmt *stmt);
-  PassResult runOnMLFunction(MLFunction *f);
+  PassResult runOnMLFunction(MLFunction *f) override;
   using StmtWalker<ComposeAffineMaps>::walk;
 };
 
 } // end anonymous namespace
 
-MLFunctionPass *mlir::createComposeAffineMapsPass() {
+FunctionPass *mlir::createComposeAffineMapsPass() {
   return new ComposeAffineMaps();
 }