diff --git a/paddle/fluid/inference/tests/api/analyzer_int8_image_classification_tester.cc b/paddle/fluid/inference/tests/api/analyzer_int8_image_classification_tester.cc index 6bfa8a821ae8cf6ef4b1fa33d8ae790700795e2b..8f8b73044232a5cacfa3609e5f8e32ccf375d418 100644 --- a/paddle/fluid/inference/tests/api/analyzer_int8_image_classification_tester.cc +++ b/paddle/fluid/inference/tests/api/analyzer_int8_image_classification_tester.cc @@ -47,6 +47,10 @@ TEST(Analyzer_int8_image_classification, quantization) { std::shared_ptr> warmup_data = paddle::inference::GetWarmupData(input_slots_all); + // INT8 implies FC oneDNN passes to be used + q_cfg.pass_builder()->AppendPass("fc_mkldnn_pass"); + q_cfg.pass_builder()->AppendPass("fc_act_mkldnn_fuse_pass"); + // configure quantizer q_cfg.EnableMkldnnQuantizer(); q_cfg.mkldnn_quantizer_config()->SetWarmupData(warmup_data);