src/cpu_gemm.c
@@ -1,4 +1,4 @@ #include "mini_blas.h" //#include "mini_blas.h" void cpu_gemm_nn(int TA, int TB, int M, int N, int K, float ALPHA, float *A, int lda, @@ -74,7 +74,12 @@ float BETA, float *C, int ldc) { // Assume beta = 1 LULZ int i, j; for(i = 0; i < M; ++i){ for(j = 0; j < N; ++j){ C[i*ldc + j] *= BETA; } } if(!TA && !TB) cpu_gemm_nn( TA, TB, M, N, K, ALPHA,A,lda, B, ldb,BETA,C,ldc); else if(TA && !TB)