From 5e21599549a4b20d285e3e94292966e0a69b65fa Mon Sep 17 00:00:00 2001 From: Shixiaowei02 Date: Thu, 27 Jun 2019 01:21:48 +0800 Subject: [PATCH] update --- paddle/fluid/lite/kernels/arm/conv_compute_test.cc | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/lite/kernels/arm/conv_compute_test.cc b/paddle/fluid/lite/kernels/arm/conv_compute_test.cc index 5253a895892..3f5ab393099 100644 --- a/paddle/fluid/lite/kernels/arm/conv_compute_test.cc +++ b/paddle/fluid/lite/kernels/arm/conv_compute_test.cc @@ -402,6 +402,7 @@ TEST(conv_arm_int8, int8_fp32) { param.output = &output_int8; param.input_scale = in_scale[0]; param.output_scale = 1; + /* std::vector w_scale_for_int8; for (auto ws : w_scale) { ws *= param.input_scale; @@ -409,7 +410,8 @@ TEST(conv_arm_int8, int8_fp32) { w_scale_for_int8.push_back(ws); } param.weight_scale = w_scale_for_int8; - + */ + param.weight_scale = w_scale; std::unique_ptr ctx_int8( new KernelContext); lite::arm::math::GemmLikeConvInt8 @@ -436,13 +438,15 @@ TEST(conv_arm_int8, int8_fp32) { param.output = &output_fp32; param.input_scale = in_scale[0]; param.output_scale = 1; + /* std::vector w_scale_for_fp32; for (auto ws : w_scale) { ws *= param.input_scale; w_scale_for_fp32.push_back(ws); } param.weight_scale = w_scale_for_fp32; - + */ + param.weight_scale = w_scale; std::unique_ptr ctx_fp32( new KernelContext); lite::arm::math::GemmLikeConvInt8 -- GitLab