Function reference_cpu_gemm

include/reference.h:6–46

template<class AType, class BType, class CType, class AccumType>
void reference_cpu_gemm(GemmParams<AType, BType, CType, AccumType> params) {
    /// pointers in params are guaranteed to be in cpu memory
    int M = params.M;
    int N = params.N;
    int K = params.K;
    const AType* hA = params.A;
    const BType* hB = params.B;
    CType* golden = params.C;
#pragma omp parallel for
    for (int i = 0; i < M; i += 64)
    {
#pragma omp parallel for
        for (int j = 0; j < N; j += 64)
        {
            AccumType accum[64 * 64] = {0};
            for (int k = 0; k < K; k += 32)
            {
                for (int kk = 0; kk < 32; ++kk)
                {
                    if (k + kk >= K) break;
                    for (int jj = 0; jj < 64; ++jj)
                    {
                        for (int ii = 0; ii < 64; ++ii)
                        {
                            accum[ii * 64 + jj] += ((AccumType)hA[(i + ii) % M * K + k + kk] * (AccumType)hB[(j + jj) % N * K + k + kk]);
                        }
                    }
                }
            }
            for (int ii = 0; ii < 64; ++ii)
            {
                for (int jj = 0; jj < 64; ++jj)
                {
                    if (i + ii < M && j + jj < N) {
                        golden[(i + ii) * N + j + jj] = (CType)(accum[ii * 64 + jj] * params.alpha + params.beta);
                    }
                }
            }
        }
    }
}


template<class AType, class BType, class CType, class AccumType,