diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp index 1ffe8ab..24b8054 100644 --- a/dabnn/layers/BinConv.cpp +++ b/dabnn/layers/BinConv.cpp @@ -87,7 +87,7 @@ bool BinConv::direct_conv_compatible() const { bool BinConv::gemm_compatible() const { #ifdef __aarch64__ - return weight_mat->h * weight_mat->n * weight_mat->c % 2 == 0; + return weight_mat->h * weight_mat->w * weight_mat->c % 2 == 0; #else return false; #endif