Edmond Yoo
2018-10-13 c59db54775606349f6ba5c6cab7fcb34498bb31d
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)