diff options
| author | Yuanqiang Liu <liuyuanqiang.yqliu@bytedance.com> | 2024-12-05 17:31:04 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-12-05 10:31:04 +0100 |
| commit | 2e51e150e161bd5fb5b8adb8655744a672ced002 (patch) | |
| tree | 62f1b1d17f1203ccda3bf31988c5dbd6becf94a2 /mlir/lib/CAPI/IR | |
| parent | a2acb2ff8b5307bb6b973820c4ededf1ddc49bb2 (diff) | |
[MLIR][Python] enhance python ir printing with pringing flags (#117836)
Close https://github.com/llvm/llvm-project/pull/65854
Diffstat (limited to 'mlir/lib/CAPI/IR')
| -rw-r--r-- | mlir/lib/CAPI/IR/Pass.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/mlir/lib/CAPI/IR/Pass.cpp b/mlir/lib/CAPI/IR/Pass.cpp index 01151eafeb52..883b7e8bb832 100644 --- a/mlir/lib/CAPI/IR/Pass.cpp +++ b/mlir/lib/CAPI/IR/Pass.cpp @@ -49,6 +49,7 @@ void mlirPassManagerEnableIRPrinting(MlirPassManager passManager, bool printModuleScope, bool printAfterOnlyOnChange, bool printAfterOnlyOnFailure, + MlirOpPrintingFlags flags, MlirStringRef treePrintingPath) { auto shouldPrintBeforePass = [printBeforeAll](Pass *, Operation *) { return printBeforeAll; @@ -60,13 +61,14 @@ void mlirPassManagerEnableIRPrinting(MlirPassManager passManager, return unwrap(passManager) ->enableIRPrinting(shouldPrintBeforePass, shouldPrintAfterPass, printModuleScope, printAfterOnlyOnChange, - printAfterOnlyOnFailure); + printAfterOnlyOnFailure, /*out=*/llvm::errs(), + *unwrap(flags)); unwrap(passManager) ->enableIRPrintingToFileTree(shouldPrintBeforePass, shouldPrintAfterPass, printModuleScope, printAfterOnlyOnChange, printAfterOnlyOnFailure, - unwrap(treePrintingPath)); + unwrap(treePrintingPath), *unwrap(flags)); } void mlirPassManagerEnableVerifier(MlirPassManager passManager, bool enable) { |