Skip to content

Commit 432a4c2

Browse files
committed
gemm_accum16() doesn't need a multiple of 16 columns (just lines).
1 parent 6696d2c commit 432a4c2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/nnet.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ static void gemm_accum16(float *out, const float *weights, int rows, int cols, i
317317
static void gemm_accum(float *out, const float *weights, int rows, int cols, int col_stride, const float *x)
318318
{
319319
int i, j;
320-
if (rows % 16 == 0 && cols % 16 == 0)
320+
if (rows % 16 == 0)
321321
{
322322
gemm_accum16(out, weights, rows, cols, col_stride, x);
323323
} else {

0 commit comments

Comments
 (0)