summaryrefslogtreecommitdiffstats
path: root/mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp')
-rw-r--r--mlir/lib/Quantizer/Transforms/AddDefaultStatsTestPass.cpp3
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>();
}
OpenPOWER on IntegriCloud