From 5eda6d2ba2f0ae4f500f744fe3ac4b6542eb5650 Mon Sep 17 00:00:00 2001 From: daquexian Date: Wed, 29 May 2019 16:59:10 +0800 Subject: [PATCH] Fix the wrong gemm_compatible() --- 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 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