From 432a4c2651ea3d432022f22fc6e01424373d82cc Mon Sep 17 00:00:00 2001 From: Jean-Marc Valin Date: Thu, 29 Nov 2018 19:50:09 -0500 Subject: [PATCH] gemm_accum16() doesn't need a multiple of 16 columns (just lines). --- src/nnet.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/nnet.c b/src/nnet.c index 6090bea..cf7d4e1 100644 --- a/src/nnet.c +++ b/src/nnet.c @@ -317,7 +317,7 @@ static void gemm_accum16(float *out, const float *weights, int rows, int cols, i static void gemm_accum(float *out, const float *weights, int rows, int cols, int col_stride, const float *x) { int i, j; - if (rows % 16 == 0 && cols % 16 == 0) + if (rows % 16 == 0) { gemm_accum16(out, weights, rows, cols, col_stride, x); } else {