intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-02-05 13:21:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

[mlir] Support passing ostream as argument for the create function.

Browse Source 
The constructor already supports passing an ostream as argument,
so let's make the create function support it too.

Differential Revision: https://reviews.llvm.org/D127449
This commit is contained in:
Okwan Kwon
2022-06-09 15:11:36 -07:00
parent f2f8ce9699
commit 5ccb9df3ba
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlir/include/mlir/Transforms/Passes.h
View File

@@ -60,7 +60,7 @@ std::unique_ptr<Pass> createStripDebugInfoPass();
/// Creates a pass which prints the list of ops and the number of occurrences in
/// the module.
std::unique_ptr<Pass> createPrintOpStatsPass();
std::unique_ptr<Pass> createPrintOpStatsPass(raw_ostream &os = llvm::errs());
/// Creates a pass which inlines calls and callable operations as defined by
/// the CallGraph.

6
mlir/lib/Transforms/OpStats.cpp
View File

@@ -19,7 +19,7 @@ using namespace mlir;
namespace {
struct PrintOpStatsPass : public PrintOpStatsBase<PrintOpStatsPass> {
explicit PrintOpStatsPass(raw_ostream &os = llvm::errs()) : os(os) {}
explicit PrintOpStatsPass(raw_ostream &os) : os(os) {}
// Prints the resultant operation statistics post iterating over the module.
void runOnOperation() override;
@@ -80,6 +80,6 @@ void PrintOpStatsPass::printSummary() {
}
}
std::unique_ptr<Pass> mlir::createPrintOpStatsPass() {
return std::make_unique<PrintOpStatsPass>();
std::unique_ptr<Pass> mlir::createPrintOpStatsPass(raw_ostream &os) {
return std::make_unique<PrintOpStatsPass>(os);
}