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;