diff --git a/paddle/fluid/lite/kernels/arm/conv_compute_test.cc b/paddle/fluid/lite/kernels/arm/conv_compute_test.cc index 5253a89589226515f6ebb8027a9318459b2db6ed..3f5ab393099be3ca37133a7655be0ffa06b6d668 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