hcore_zgemmbdcd.c

Go to the documentation of this file.

#include "coreblas/coreblas.h"
#include "coreblas/lapacke.h"
#include <assert.h>

//FIXME PREVIOUS DECLARION OF CBLAS_SADDR ~/hicma/chameleon/build/include/chameleon/coreblas/include/coreblas.h
#undef CBLAS_SADDR
#define CBLAS_SADDR(_val) (_val)

int gemmbdcd_print_index = 0;
int gemmbdcd_print_mat = 0;

/*
 * CD=AU*BV'. 
 * Rank of tile AU is in Ark.
 * Rank of tile BV is in Brk.
 * Multiplied tiles must have same rank.
 * CD is dense output.
*/
void HCORE_zgemmbdcd(MORSE_enum transA, MORSE_enum transB,
        int M, int N,
        double alpha, 
        double *AU, 
        double *AV, 
        double *Ark, 
        int LDA,
        double *BD, 
        int LDB,
        double beta, 
        double *CD, 
        int LDC,
        double *work 
)
{
    /*printf("%d|M:%d N:%d K:%d LDA:%d LDB:%d LDC:%d rk:%d acc:%e a:%e b:%e\n",*/
            /*__LINE__, M, N, K, LDA, LDB, LDC, rk, acc, alpha, beta);*/

    /*
     * NOTES:
     * assumptions on matrix dimensions are marked as //ASSUMPTION
     * I am currently allocating and freeing temporary buffers.
     * They are marked as //ALLOCATE
     */
    /*printf("%d %d|%g->%d. %g %g, %g %g\n",  */
            /*__LINE__, __COUNTER__,*/
            /*Crk[0], new_Crk, CU[0], CU[1], CV[0], CV[1]);*/
    //hcore_dgemm(Aij, ik, Ajk, -1, rank, acc); 
    int64_t _Ark = (int64_t)(Ark[0]);
    int64_t sizeofAVBD = LDB * N * sizeof(double); 
    if(gemmbdcd_print_index){
        printf("Ark:%d M:%d N:%d K:%d  LDA:%d LDB:%d LDC:%d sizeofAVBD=%d transA:%d transB:%d alpha:%g beta:%g\n",
                _Ark, M, N, _Ark, LDA, LDB, LDC, sizeofAVBD, transA, transB, alpha, beta  );
    }
    double one = 1.0, zero = 0.0, minusone = -1.0;
    if(gemmbdcd_print_mat){
        /*hcfrk_printmat(AU,  M, _Ark, LDA); */
        /*hcfrk_printmat(BV,  N, _Brk, LDB); */
    }
    double *AVBD = malloc(sizeofAVBD);
    if ( transA == MorseTrans) {
        /*printf("AUBD=AU^TxBD m,n,k:%d,%d,%d ld:%d,%d,%d\n", _Ark, N, M, LDA, LDB, LDB);*/
        cblas_dgemm(
                CblasColMajor,
                CblasTrans, CblasNoTrans,
                _Ark, N, M,
                CBLAS_SADDR(one),  AV,  LDA,
                BD,  LDB,
                CBLAS_SADDR(zero), AVBD, LDB);
        /*printf("CD=AVxAUBD m,n,k:%d,%d,%d ld:%d,%d,%d\n", LDA, N, _Ark, LDA, LDB, LDC);*/
        cblas_dgemm(
                CblasColMajor,
                CblasNoTrans, CblasNoTrans,
                LDA, N, _Ark,
                CBLAS_SADDR(alpha),  AU,  LDA,
                AVBD, LDB,
                CBLAS_SADDR(beta), CD, LDC);
    } else {
        /*printf("AVBD=AV^TxBD m,n,k:%d,%d,%d ld:%d,%d,%d\n", _Ark, N, LDB, LDA, LDB, LDB);*/
        cblas_dgemm(
                CblasColMajor,
                CblasTrans, CblasNoTrans,
                _Ark, N, LDB,
                CBLAS_SADDR(one),  AV,  LDA,
                BD,  LDB,
                CBLAS_SADDR(zero), AVBD, LDB);
        /*printf("CD=AUxAVBD m,n,k:%d,%d,%d ld:%d,%d,%d\n", M, N, _Ark, LDA, LDB, LDC);*/
        cblas_dgemm(
                CblasColMajor,
                CblasNoTrans, CblasNoTrans,
                M, N, _Ark,
                CBLAS_SADDR(alpha),  AU,  LDA,
                AVBD, LDB,
                CBLAS_SADDR(beta), CD, LDC);
    }
    free(AVBD);
    if(gemmbdcd_print_mat){
        /*hcfrk_printmat(CD,  M, N, LDC); */
    }

}