dot.h

Go to the documentation of this file.

/*
Copyright 2020 Peter Rakyta, Ph.D.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

#ifndef Dot_H
#define Dot_H


#include "matrix.h"
#include "logging.h"
#include "matrix_float.h"

#ifndef CPYTHON
#include <tbb/tbb.h>
#endif

#ifdef __cplusplus
extern "C"
{
#endif

typedef enum CBLAS_ORDER     {CblasRowMajor=101, CblasColMajor=102} CBLAS_ORDER;
typedef enum CBLAS_TRANSPOSE {CblasNoTrans=111, CblasTrans=112, CblasConjTrans=113, CblasConjNoTrans=114} CBLAS_TRANSPOSE;

void cblas_zgemm(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);

void cblas_cgemm(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);

void cblas_zgemv(const enum CBLAS_ORDER Order, const enum CBLAS_TRANSPOSE TransA, const int M, const int N, const void *alpha, const void *A, const int lda,
                 const void *X, const int incX, const void *beta, void *Y, const int incY);


#if BLAS==1 // MKL

void vzConj(int num, QGD_Complex16* input, QGD_Complex16* output);

#endif

#ifdef __cplusplus
}
#endif





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

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



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



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

// relieve Python extension from TBB functionalities
#ifndef CPYTHON

struct row_indices {

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

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




class zgemm_Task_serial : public logging {

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




public:


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

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

void zgemm_chunk();

}; // zgemm_Task_serial



class zgemm_Task: public zgemm_Task_serial {



public:


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

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



void execute(tbb::task_group &g);



}; // zgemm_Task


#endif // CPYTHON


#endif //Dot_H