dot.h

Go to the documentation of this file.

#ifndef Dot_H
#define Dot_H

#ifdef __cplusplus
extern "C"
{
#endif
#include <gsl/gsl_blas_types.h>
#ifdef __cplusplus
}
#endif

#ifndef CblasConjNoTrans
#define CblasConjNoTrans 114
#endif


#include "qgd/matrix.h"

#ifdef CBLAS
#ifdef __cplusplus
extern "C"
{
#endif


void cblas_zgemm3m(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_TRANSPOSE TransB, const int M, const int N, const int K,
           const void *alpha, const void *A, const int lda, const void *B, const int ldb, const void *beta, void *C, const int ldc);


#if BLAS==1 // MKL

void vzConj(int num, pic::Complex16* input, pic::Complex16* output);

#endif

#ifdef __cplusplus
}
#endif
#endif



Matrix dot( Matrix &A, Matrix &B );



bool check_matrices( Matrix &A, Matrix &B );



void get_cblas_transpose( Matrix &A, CBLAS_TRANSPOSE &transpose );

struct row_indices {

    size_t Arows_start;
    size_t Arows_end;
    size_t Arows;
    size_t Brows_start;
    size_t Brows_end;
    size_t Brows;
    size_t Crows_start;
    size_t Crows_end;
    size_t Crows;
};

struct col_indices {
    size_t Acols_start;
    size_t Acols_end;
    size_t Acols;
    size_t Bcols_start;
    size_t Bcols_end;
    size_t Bcols;
    size_t Ccols_start;
    size_t Ccols_end;
    size_t Ccols;
};


class Cont_Task_rowsA: public tbb::task {

public:

Cont_Task_rowsA();

tbb::task* execute();

};


class zgemm3m_Task_serial {

public:
    Matrix A;
    Matrix B;
    Matrix C;
    row_indices rows;
    col_indices cols;
    CBLAS_ORDER order;




public:

zgemm3m_Task_serial( Matrix &A_in, Matrix &B_in, Matrix &C_in);

zgemm3m_Task_serial( Matrix &A_in, Matrix &B_in, Matrix &C_in, row_indices& rows_in, col_indices& cols_in);

void zgemm3m_chunk();

}; // zgemm3m_Task_serial



class zgemm3m_Task: public tbb::task, public zgemm3m_Task_serial {


public:
    // reuse the constructors of class zgemm3m_Task_serial
    using zgemm3m_Task_serial::zgemm3m_Task_serial;


tbb::task* execute();



}; // zgemm3m_Task



#endif