gemm_accum16() doesn't need a multiple of 16 columns (just lines).

2018-11-29 19:50:09 -05:00 · 2018-11-29 19:50:09 -05:00 · 432a4c2651
parent 6696d2c355
commit 432a4c2651
1 changed files with 1 additions and 1 deletions
--- 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 {