diff options
Diffstat (limited to 'mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp')
-rw-r--r-- | mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp | 3 |
1 files changed, 2 insertions, 1 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>(); } |