From 3e8396295bab056f082452d1e04a74edec034ed5 Mon Sep 17 00:00:00 2001 From: HappyAngel Date: Sat, 30 Nov 2019 11:12:20 +0800 Subject: [PATCH] fiix conv_bn fuse error in model_test, test=develop (#2531) --- lite/api/benchmark.cc | 5 +---- lite/api/model_test.cc | 5 +---- 2 files changed, 2 insertions(+), 8 deletions(-) diff --git a/lite/api/benchmark.cc b/lite/api/benchmark.cc index c137324b57..50a2797d0f 100644 --- a/lite/api/benchmark.cc +++ b/lite/api/benchmark.cc @@ -44,10 +44,7 @@ void OutputOptModel(const std::string& load_model_dir, const std::vector>& input_shapes) { lite_api::CxxConfig config; config.set_model_dir(load_model_dir); - std::vector vaild_places = { - Place{TARGET(kARM), PRECISION(kFloat)}, - Place{TARGET(kX86), PRECISION(kFloat)}, - }; + std::vector vaild_places = {Place{TARGET(kARM), PRECISION(kFloat)}}; if (FLAGS_is_quantized_model) { vaild_places.insert(vaild_places.begin(), Place{TARGET(kARM), PRECISION(kInt8)}); diff --git a/lite/api/model_test.cc b/lite/api/model_test.cc index 1358267000..aadb8df755 100644 --- a/lite/api/model_test.cc +++ b/lite/api/model_test.cc @@ -46,10 +46,7 @@ void OutputOptModel(const std::string& load_model_dir, const std::vector>& input_shapes) { lite_api::CxxConfig config; config.set_model_dir(load_model_dir); - config.set_valid_places({ - Place{TARGET(kX86), PRECISION(kFloat)}, - Place{TARGET(kARM), PRECISION(kFloat)}, - }); + config.set_valid_places({Place{TARGET(kARM), PRECISION(kFloat)}}); auto predictor = lite_api::CreatePaddlePredictor(config); // delete old optimized model -- GitLab