From de3f85ff3f888a52ae49da365155bda2e24e0ca5 Mon Sep 17 00:00:00 2001 From: daquexian Date: Wed, 29 May 2019 17:00:47 +0800 Subject: [PATCH] Enable bgemm for armv7 --- dabnn/layers/BinConv.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp index 91ad541..e6bdf79 100644 --- 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;