From 3966bb08b383c830460332a3f60c3c3da55855f2 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 1 Jun 2020 18:53:28 +0800
Subject: [PATCH] feat(dnn/test): split cpu.convolution

GitOrigin-RevId: fa28d3d760c230e8d68975981c0cb40f1c46fb19
---
 dnn/test/cpu/convolution.cpp | 37 ++++++++++++++++++++++++++++++++++--
 1 file changed, 35 insertions(+), 2 deletions(-)

diff --git a/dnn/test/cpu/convolution.cpp b/dnn/test/cpu/convolution.cpp
index 38db86ceb..b7226fd5b 100644
--- a/dnn/test/cpu/convolution.cpp
+++ b/dnn/test/cpu/convolution.cpp
@@ -26,7 +26,7 @@ Convolution::Param gconv_param(Convolution::Param p) {
 
 } // anonymous namespace
 
-#define CONVOLUTION_ARG_DIV_SIZE 230
+#define CONVOLUTION_ARG_DIV_SIZE 100
 TEST_F(CPU, CONVOLUTION_0) {
     using namespace convolution;
     std::vector<TestArg> args = get_args();
@@ -39,18 +39,51 @@ TEST_F(CPU, CONVOLUTION_0) {
     }
 }
 
+#define CONVOLUTION1_ARG_LOOP_END_TIME (CONVOLUTION_ARG_DIV_SIZE + 205)
+
 TEST_F(CPU, CONVOLUTION_1) {
     using namespace convolution;
     std::vector<TestArg> args = get_args();
     auto loop_size = args.size();
     ASSERT_GT(loop_size, CONVOLUTION_ARG_DIV_SIZE);
+    ASSERT_GT(loop_size, CONVOLUTION1_ARG_LOOP_END_TIME);
+    Checker<Convolution> checker(handle());
+    for (unsigned int i = CONVOLUTION_ARG_DIV_SIZE;
+         i < CONVOLUTION1_ARG_LOOP_END_TIME; i++) {
+        checker.set_param(args[i].param)
+                .execs({args[i].src, args[i].filter, {}});
+    }
+}
+
+#define CONVOLUTION2_ARG_LOOP_END_TIME (CONVOLUTION1_ARG_LOOP_END_TIME + 200)
+TEST_F(CPU, CONVOLUTION_2) {
+    using namespace convolution;
+    std::vector<TestArg> args = get_args();
+    auto loop_size = args.size();
+    ASSERT_GT(loop_size, CONVOLUTION2_ARG_LOOP_END_TIME);
     Checker<Convolution> checker(handle());
-    for (unsigned int i = CONVOLUTION_ARG_DIV_SIZE; i < loop_size; i++) {
+    for (unsigned int i = CONVOLUTION1_ARG_LOOP_END_TIME;
+         i < CONVOLUTION2_ARG_LOOP_END_TIME; i++) {
         checker.set_param(args[i].param)
                 .execs({args[i].src, args[i].filter, {}});
     }
 }
+
+TEST_F(CPU, CONVOLUTION_3) {
+    using namespace convolution;
+    std::vector<TestArg> args = get_args();
+    auto loop_size = args.size();
+    ASSERT_GT(loop_size, CONVOLUTION2_ARG_LOOP_END_TIME);
+    Checker<Convolution> checker(handle());
+    for (unsigned int i = CONVOLUTION2_ARG_LOOP_END_TIME; i < loop_size; i++) {
+        checker.set_param(args[i].param)
+                .execs({args[i].src, args[i].filter, {}});
+    }
+}
+
 #undef CONVOLUTION_ARG_DIV_SIZE
+#undef CONVOLUTION1_ARG_LOOP_END_TIME
+#undef CONVOLUTION2_ARG_LOOP_END_TIME
 
 #define CB_CONV_CONFIG_COMBINATIONS(KSIZE)                                \
     TEST_F(CPU, CONV_CONFIG_COMBINATIONS_KSIZE_1_KSIZE_##KSIZE) {         \
-- 
GitLab