Disable bgemm when weight_mat->c == 1 in armv7

JDAI-CV · Aug 21, 2019 · ce6e843 · ce6e843
1 parent 934d215
commit ce6e843
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp
@@ -106,7 +106,15 @@ bool BinConv::direct_conv_compatible() const {
 
 bool BinConv::gemm_compatible() const {
 #ifdef __ARM_NEON
+#ifdef __aarch64__
     return true;
+#else
+    // If weight_mat->c == 1 (weight_mat has 64 channels), we use bconv_64
+    // in aarch64 for the fastest speed, however, bconv_64 is not implemented
+    // in armv7
+    // TODO: Implement bconv_64 for armv7
+    return weight_mat->c != 1;
+#endif
 #else
     return false;
 #endif