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

fix ut

上级 015a408a
......@@ -326,8 +326,8 @@ void test_conv_int8(const std::vector<DDim>& 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<DDim>& 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<DDim> 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}) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册