Change the return type of createPrintCFGGraphPass to match other passes.

PiperOrigin-RevId: 271252404
llvm · Sep 26, 2019 · 47a7021 · 47a7021
1 parent ae13c28
commit 47a7021
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/mlir/include/mlir/Transforms/ViewRegionGraph.h b/mlir/include/mlir/Transforms/ViewRegionGraph.h
@@ -41,7 +41,7 @@ llvm::raw_ostream &writeGraph(llvm::raw_ostream &os, Region &region,
                               bool shortNames = false, const Twine &title = "");
 
 /// Creates a pass to print CFG graphs.
-OpPassBase<FuncOp> *
+std::unique_ptr<mlir::OpPassBase<mlir::FuncOp>>
 createPrintCFGGraphPass(llvm::raw_ostream &os = llvm::errs(),
                         bool shortNames = false, const llvm::Twine &title = "");
 

diff --git a/mlir/lib/Transforms/ViewRegionGraph.cpp b/mlir/lib/Transforms/ViewRegionGraph.cpp
@@ -85,10 +85,10 @@ struct PrintCFGPass : public FunctionPass<PrintCFGPass> {
 };
 } // namespace
 
-OpPassBase<FuncOp> *mlir::createPrintCFGGraphPass(llvm::raw_ostream &os,
-                                                  bool shortNames,
-                                                  const llvm::Twine &title) {
-  return new PrintCFGPass(os, shortNames, title);
+std::unique_ptr<mlir::OpPassBase<mlir::FuncOp>>
+mlir::createPrintCFGGraphPass(llvm::raw_ostream &os, bool shortNames,
+                              const llvm::Twine &title) {
+  return std::make_unique<PrintCFGPass>(os, shortNames, title);
 }
 
 static PassRegistration<PrintCFGPass> pass("print-cfg-graph",