From eae7b34cc0906268ddef2e64551ea5ca0951c0c8 Mon Sep 17 00:00:00 2001 From: Ruslan_Lenin <9992012@gmail.com> Date: Fri, 29 Jun 2018 11:39:03 +0300 Subject: [PATCH] Update gemm.c optimization calculations --- src/gemm.c | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/gemm.c b/src/gemm.c index 317fd650..506687be 100644 --- a/src/gemm.c +++ b/src/gemm.c @@ -273,10 +273,12 @@ void gemm_cpu(int TA, int TB, int M, int N, int K, float ALPHA, float *C, int ldc) { //printf("cpu: %d %d %d %d %d %f %d %d %f %d\n",TA, TB, M, N, K, ALPHA, lda, ldb, BETA, ldc); - int i, j; - for(i = 0; i < M; ++i){ - for(j = 0; j < N; ++j){ - C[i*ldc + j] *= BETA; + if (BETA != 1){ + int i, j; + for(i = 0; i < M; ++i){ + for(j = 0; j < N; ++j){ + C[i*ldc + j] *= BETA; + } } }