From 6808a8212728968e30db598d448fecf0209cf3f4 Mon Sep 17 00:00:00 2001 From: Sylwester Fraczek Date: Wed, 27 Jul 2022 21:21:59 +0800 Subject: [PATCH] enable add passes pre-calculating scales/quantizing weights --- paddle/fluid/inference/api/paddle_pass_builder.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/fluid/inference/api/paddle_pass_builder.cc b/paddle/fluid/inference/api/paddle_pass_builder.cc index 67acbd7e51179..bd390619f7255 100644 --- a/paddle/fluid/inference/api/paddle_pass_builder.cc +++ b/paddle/fluid/inference/api/paddle_pass_builder.cc @@ -405,6 +405,8 @@ void CpuPassStrategy::EnableMkldnnInt8() { passes_.push_back("cpu_quantize_pass"); passes_.push_back("cpu_quantize_squash_pass"); passes_.push_back("simplify_with_basic_ops_pass"); + passes_.push_back("int8_scale_calculation_mkldnn_pass"); + passes_.push_back("params_quantization_mkldnn_pass"); passes_.push_back("mkldnn_inplace_pass"); passes_.push_back("runtime_context_cache_pass"); }