From f1b100c77ba005899c60f3dea74607d5daad3f52 Mon Sep 17 00:00:00 2001
From: River Riddle <riverriddle@google.com>
Date: Fri, 13 Sep 2019 13:33:46 -0700
Subject: NFC: Finish replacing FunctionPassBase/ModulePassBase with
 OpPassBase.

These directives were temporary during the generalization of FunctionPass/ModulePass to OpPass.

PiperOrigin-RevId: 268970259
---
 mlir/lib/Transforms/ViewOpGraph.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'mlir/lib/Transforms/ViewOpGraph.cpp')

diff --git a/mlir/lib/Transforms/ViewOpGraph.cpp b/mlir/lib/Transforms/ViewOpGraph.cpp
index afb65c7d148..7f65a143a96 100644
--- a/mlir/lib/Transforms/ViewOpGraph.cpp
+++ b/mlir/lib/Transforms/ViewOpGraph.cpp
@@ -153,7 +153,7 @@ llvm::raw_ostream &mlir::writeGraph(llvm::raw_ostream &os, mlir::Block &block,
   return llvm::WriteGraph(os, &block, shortNames, title);
 }
 
-std::unique_ptr<mlir::ModulePassBase>
+std::unique_ptr<mlir::OpPassBase<mlir::ModuleOp>>
 mlir::createPrintOpGraphPass(llvm::raw_ostream &os, bool shortNames,
                              const llvm::Twine &title) {
   return std::make_unique<PrintOpPass>(os, shortNames, title);
-- 
cgit v1.2.3