From 0668b343f0d4e47c1a5c18dc1635a87b8f31227b Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 7 Apr 2020 21:54:27 +0800
Subject: [PATCH] feat(dnn/arm_common): add arm_common int8 channel wise stride
 1

GitOrigin-RevId: 679916a986ad6aff22821148179cb17e32b5bfc4
---
 dnn/src/fallback/conv_bias/common.h | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/dnn/src/fallback/conv_bias/common.h b/dnn/src/fallback/conv_bias/common.h
index aac5ce72..29fdae8c 100644
--- a/dnn/src/fallback/conv_bias/common.h
+++ b/dnn/src/fallback/conv_bias/common.h
@@ -122,6 +122,22 @@ using BiasMode = ConvBiasForward::BiasMode;
             break;                            \
     }
 
+#define DISPATCH_FILTER_CHANNEL_WISE(filter, kern, arg...) \
+    switch (filter) {                                      \
+        case 2:                                            \
+            kern(2, ##arg);                                \
+            break;                                         \
+        case 3:                                            \
+            kern(3, ##arg);                                \
+            break;                                         \
+        case 5:                                            \
+            kern(5, ##arg);                                \
+            break;                                         \
+        default:                                           \
+            megdnn_assert(0);                              \
+            break;                                         \
+    }
+
 enum class PostprocessMode : uint8_t {
     FLOAT = 0,  ///< support all biasmode and no_nonlinemode
     NO_PROCESS, ///<support  non bias and identity
-- 
GitLab