1 files changed, 2 insertions, 2 deletions
diff --git a/mlir/lib/Transforms/LoopUnrollAndJam.cpp b/mlir/lib/Transforms/LoopUnrollAndJam.cpp
index a4a69ecdc5c..b05063232ed 100644
--- a/mlir/lib/Transforms/LoopUnrollAndJam.cpp
+++ b/mlir/lib/Transforms/LoopUnrollAndJam.cpp
@@ -66,7 +66,7 @@ static llvm::cl::opt<unsigned>
 namespace {
 /// Loop unroll jam pass. Currently, this just unroll jams the first
 /// outer loop in an MLFunction.
-struct LoopUnrollAndJam : public MLFunctionPass {
+struct LoopUnrollAndJam : public FunctionPass {
   Optional<unsigned> unrollJamFactor;
   static const unsigned kDefaultUnrollJamFactor = 4;
 
@@ -78,7 +78,7 @@ struct LoopUnrollAndJam : public MLFunctionPass {
 };
 } // end anonymous namespace
 
-MLFunctionPass *mlir::createLoopUnrollAndJamPass(int unrollJamFactor) {
+FunctionPass *mlir::createLoopUnrollAndJamPass(int unrollJamFactor) {
   return new LoopUnrollAndJam(
       unrollJamFactor == -1 ? None : Optional<unsigned>(unrollJamFactor));
 }