diff options
Diffstat (limited to 'mlir/lib/Quantizer/Transforms')
3 files changed, 5 insertions, 3 deletions
diff --git a/mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp b/mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp index a2d38ce211d..696c1e2db3a 100644 --- a/mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp +++ b/mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp @@ -118,7 +118,8 @@ void AddDefaultStatsPass::runWithConfig(SolverContext &solverContext, }); } -std::unique_ptr<FunctionPassBase> mlir::quantizer::createAddDefaultStatsPass() { +std::unique_ptr<OpPassBase<FuncOp>> +mlir::quantizer::createAddDefaultStatsPass() { return std::make_unique<AddDefaultStatsPass>(); } diff --git a/mlir/lib/Quantizer/Transforms/InferQuantizedTypesPass.cpp b/mlir/lib/Quantizer/Transforms/InferQuantizedTypesPass.cpp index ff293fc93aa..7c449e32c4c 100644 --- a/mlir/lib/Quantizer/Transforms/InferQuantizedTypesPass.cpp +++ b/mlir/lib/Quantizer/Transforms/InferQuantizedTypesPass.cpp @@ -286,7 +286,8 @@ void InferQuantizedTypesPass::transformResultType(CAGResultAnchor *anchor, } } -std::unique_ptr<ModulePassBase> mlir::quantizer::createInferQuantizedTypesPass( +std::unique_ptr<OpPassBase<ModuleOp>> +mlir::quantizer::createInferQuantizedTypesPass( SolverContext &solverContext, const TargetConfiguration &config) { return std::make_unique<InferQuantizedTypesPass>(solverContext, config); } diff --git a/mlir/lib/Quantizer/Transforms/RemoveInstrumentationPass.cpp b/mlir/lib/Quantizer/Transforms/RemoveInstrumentationPass.cpp index b9fbf27d24f..0266520bec3 100644 --- a/mlir/lib/Quantizer/Transforms/RemoveInstrumentationPass.cpp +++ b/mlir/lib/Quantizer/Transforms/RemoveInstrumentationPass.cpp @@ -66,7 +66,7 @@ void RemoveInstrumentationPass::runOnFunction() { applyPatternsGreedily(func, patterns); } -std::unique_ptr<FunctionPassBase> +std::unique_ptr<OpPassBase<FuncOp>> mlir::quantizer::createRemoveInstrumentationPass() { return std::make_unique<RemoveInstrumentationPass>(); } |
