Enable bgemm for armv7

JDAI-CV · May 29, 2019 · de3f85f · de3f85f
1 parent 5af2ec3
commit de3f85f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp
@@ -86,7 +86,7 @@ bool BinConv::direct_conv_compatible() const {
 }
 
 bool BinConv::gemm_compatible() const {
-#ifdef __aarch64__
+#ifdef __ARM_NEON
     return weight_mat->h * weight_mat->w * weight_mat->c % 2 == 0;
 #else
     return false;