提交 6d92deff 编写于 作者: C chenjiaoAngel

fix ut

上级 015a408a
...@@ -327,7 +327,7 @@ void test_conv_int8(const std::vector<DDim>& input_dims, ...@@ -327,7 +327,7 @@ void test_conv_int8(const std::vector<DDim>& input_dims,
dout_basic_fp32[i] = dout_basic_fp32[i] > six ? six : dout_basic_fp32[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] * double gops = 2.0 * dim_out.production() * dim_in[1] * weight_dim[2] *
weight_dim[3] / group; weight_dim[3] / group;
/// warm up /// warm up
...@@ -484,7 +484,7 @@ void test_conv_int8(const std::vector<DDim>& input_dims, ...@@ -484,7 +484,7 @@ void test_conv_int8(const std::vector<DDim>& input_dims,
float alpha = 1.f) {} float alpha = 1.f) {}
#endif // LITE_WITH_ARM #endif // LITE_WITH_ARM
#if 0 /// 3x3dw #if 1 /// 3x3dw
TEST(TestConv3x3DWInt8, test_conv3x3_depthwise) { TEST(TestConv3x3DWInt8, test_conv3x3_depthwise) {
if (FLAGS_basic_test) { if (FLAGS_basic_test) {
for (auto& stride : {1, 2}) { for (auto& stride : {1, 2}) {
...@@ -520,7 +520,7 @@ TEST(TestConv3x3DWInt8, test_conv3x3_depthwise) { ...@@ -520,7 +520,7 @@ TEST(TestConv3x3DWInt8, test_conv3x3_depthwise) {
} }
#endif /// 3x3dw #endif /// 3x3dw
#if 0 /// 5x5dw #if 1 /// 5x5dw
TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) { TEST(TestConv5x5DWInt8, test_conv5x5_depthwise) {
if (FLAGS_basic_test) { if (FLAGS_basic_test) {
for (auto& stride : {1, 2}) { for (auto& stride : {1, 2}) {
...@@ -563,7 +563,7 @@ TEST(TestConv1x1s1Int8, test_conv1x1s1) { ...@@ -563,7 +563,7 @@ TEST(TestConv1x1s1Int8, test_conv1x1s1) {
for (auto& cout : {1, 5, 17}) { for (auto& cout : {1, 5, 17}) {
for (auto& g : {1, 2}) { for (auto& g : {1, 2}) {
for (auto& flag_bias : {false, true}) { for (auto& flag_bias : {false, true}) {
for (auto& flag_act : {0, 1, 2}) { for (auto& flag_act : {0, 1, 2, 4}) {
std::vector<DDim> dims; std::vector<DDim> dims;
if (cin % g != 0 || cout % g != 0) { if (cin % g != 0 || cout % g != 0) {
continue; continue;
...@@ -595,7 +595,7 @@ TEST(TestConv1x1s1Int8, test_conv1x1s1) { ...@@ -595,7 +595,7 @@ TEST(TestConv1x1s1Int8, test_conv1x1s1) {
} }
#endif /// conv1x1s1 #endif /// conv1x1s1
#if 0 /// conv3x3s1 #if 1 /// conv3x3s1
TEST(TestConv3x3s1Int8, test_conv_3x3s1) { TEST(TestConv3x3s1Int8, test_conv_3x3s1) {
if (FLAGS_basic_test) { if (FLAGS_basic_test) {
for (auto& cin : {1, 3, 8, 33}) { for (auto& cin : {1, 3, 8, 33}) {
...@@ -637,7 +637,7 @@ TEST(TestConv3x3s1Int8, test_conv_3x3s1) { ...@@ -637,7 +637,7 @@ TEST(TestConv3x3s1Int8, test_conv_3x3s1) {
} }
#endif /// conv3x3s1 #endif /// conv3x3s1
#if 0 /// conv3x3s2 #if 1 /// conv3x3s2
TEST(TestConv3x3s2Int8, test_conv_3x3s2) { TEST(TestConv3x3s2Int8, test_conv_3x3s2) {
if (FLAGS_basic_test) { if (FLAGS_basic_test) {
for (auto& cin : {1, 3, 31}) { for (auto& cin : {1, 3, 31}) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册