From 6d92deff5e40be6de09930b6fba0262b538f8e72 Mon Sep 17 00:00:00 2001 From: chenjiaoAngel Date: Thu, 7 May 2020 08:39:21 -0400 Subject: [PATCH] fix ut --- lite/tests/math/conv_int8_compute_test.cc | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/lite/tests/math/conv_int8_compute_test.cc b/lite/tests/math/conv_int8_compute_test.cc index 456f32b8a5..c6226660cf 100644 --- a/lite/tests/math/conv_int8_compute_test.cc +++ b/lite/tests/math/conv_int8_compute_test.cc @@ -326,8 +326,8 @@ void test_conv_int8(const std::vector& input_dims, dout_basic_int8[i] = dout_basic_int8[i] > six ? six : dout_basic_int8[i]; dout_basic_fp32[i] = dout_basic_fp32[i] > six ? six : dout_basic_fp32[i]; } + } } - double gops = 2.0 * dim_out.production() * dim_in[1] * weight_dim[2] * weight_dim[3] / group; /// warm up @@ -484,7 +484,7 @@ void test_conv_int8(const std::vector& input_dims, float alpha = 1.f) {} #endif // LITE_WITH_ARM -#if 0 /// 3x3dw +#if 1 /// 3x3dw TEST(TestConv3x3DWInt8, test_conv3x3_depthwise) { if (FLAGS_basic_test) { for (auto& stride : {1, 2}) { @@ -520,7 +520,7 @@ TEST(TestConv3x3DWInt8, test_conv3x3_depthwise) { } #endif /// 3x3dw -#if 0 /// 5x5dw +#if 1 /// 5x5dw TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) { if (FLAGS_basic_test) { for (auto& stride : {1, 2}) { @@ -563,7 +563,7 @@ TEST(TestConv1x1s1Int8, test_conv1x1s1) { for (auto& cout : {1, 5, 17}) { for (auto& g : {1, 2}) { for (auto& flag_bias : {false, true}) { - for (auto& flag_act : {0, 1, 2}) { + for (auto& flag_act : {0, 1, 2, 4}) { std::vector dims; if (cin % g != 0 || cout % g != 0) { continue; @@ -595,7 +595,7 @@ TEST(TestConv1x1s1Int8, test_conv1x1s1) { } #endif /// conv1x1s1 -#if 0 /// conv3x3s1 +#if 1 /// conv3x3s1 TEST(TestConv3x3s1Int8, test_conv_3x3s1) { if (FLAGS_basic_test) { for (auto& cin : {1, 3, 8, 33}) { @@ -637,7 +637,7 @@ TEST(TestConv3x3s1Int8, test_conv_3x3s1) { } #endif /// conv3x3s1 -#if 0 /// conv3x3s2 +#if 1 /// conv3x3s2 TEST(TestConv3x3s2Int8, test_conv_3x3s2) { if (FLAGS_basic_test) { for (auto& cin : {1, 3, 31}) { -- GitLab