Grid 0.7.0
GridBLAS Class Reference

#include <BatchedBlas.h>

Public Member Functions

 GridBLAS ()
 ~GridBLAS ()
void synchronise (void)
void gemmBatched (int m, int n, int k, ComplexD alpha, deviceVector< ComplexD * > &Amk, deviceVector< ComplexD * > &Bkn, ComplexD beta, deviceVector< ComplexD * > &Cmn)
void gemmBatched (int m, int n, int k, ComplexF alpha, deviceVector< ComplexF * > &Amk, deviceVector< ComplexF * > &Bkn, ComplexF beta, deviceVector< ComplexF * > &Cmn)
void gemmBatched (int m, int n, int k, RealD alpha, deviceVector< RealD * > &Amk, deviceVector< RealD * > &Bkn, RealD beta, deviceVector< RealD * > &Cmn)
void gemmBatched (int m, int n, int k, RealF alpha, deviceVector< RealF * > &Amk, deviceVector< RealF * > &Bkn, RealF beta, deviceVector< RealF * > &Cmn)
void gemmBatched (GridBLASOperation_t OpA, GridBLASOperation_t OpB, int m, int n, int k, ComplexD alpha, deviceVector< ComplexD * > &Amk, deviceVector< ComplexD * > &Bkn, ComplexD beta, deviceVector< ComplexD * > &Cmn)
void gemmBatched (GridBLASOperation_t OpA, GridBLASOperation_t OpB, int m, int n, int k, ComplexF alpha, deviceVector< ComplexF * > &Amk, deviceVector< ComplexF * > &Bkn, ComplexF beta, deviceVector< ComplexF * > &Cmn)
void gemmBatched (GridBLASOperation_t OpA, GridBLASOperation_t OpB, int m, int n, int k, RealF alpha, deviceVector< RealF * > &Amk, deviceVector< RealF * > &Bkn, RealF beta, deviceVector< RealF * > &Cmn)
void gemmBatched (GridBLASOperation_t OpA, GridBLASOperation_t OpB, int m, int n, int k, RealD alpha, deviceVector< RealD * > &Amk, deviceVector< RealD * > &Bkn, RealD beta, deviceVector< RealD * > &Cmn)
template<class CComplex>
double benchmark (int M, int N, int K, int BATCH)

Static Public Member Functions

static void Init (void)

Static Public Attributes

static gridblasHandle_t gridblasHandle
static int gridblasInit

Detailed Description

Definition at line 69 of file BatchedBlas.h.

Constructor & Destructor Documentation

◆ GridBLAS()

GridBLAS::GridBLAS ( )
inline

Definition at line 102 of file BatchedBlas.h.

References Init().

◆ ~GridBLAS()

GridBLAS::~GridBLAS ( )
inline

Definition at line 103 of file BatchedBlas.h.

Member Function Documentation

◆ Init()

void GridBLAS::Init ( void )
inlinestatic

Definition at line 76 of file BatchedBlas.h.

References gridblasHandle, and gridblasInit.

Referenced by GridBLAS().

◆ synchronise()

void GridBLAS::synchronise ( void )
inline

Definition at line 118 of file BatchedBlas.h.

References accelerator_barrier, and gridblasHandle.

Referenced by benchmark(), MultiRHSBlockProject< Field >::blockProject(), MultiRHSBlockProject< Field >::blockPromote(), MultiRHSDeflation< Field >::DeflateSources(), gemmBatched(), gemmBatched(), gemmBatched(), gemmBatched(), MultiRHSBlockCGLinalg< Field >::InnerProductMatrix(), MultiGeneralCoarsenedMatrix< Fobj, CComplex, nbasis >::M(), and MultiRHSBlockCGLinalg< Field >::MulMatrix().

◆ gemmBatched() [1/8]

void GridBLAS::gemmBatched ( int m,
int n,
int k,
ComplexD alpha,
deviceVector< ComplexD * > & Amk,
deviceVector< ComplexD * > & Bkn,
ComplexD beta,
deviceVector< ComplexD * > & Cmn )
inline

Definition at line 136 of file BatchedBlas.h.

References gemmBatched(), and GridBLAS_OP_N.

Referenced by benchmark(), MultiRHSBlockProject< Field >::blockProject(), MultiRHSBlockProject< Field >::blockPromote(), MultiRHSDeflation< Field >::DeflateSources(), gemmBatched(), gemmBatched(), gemmBatched(), gemmBatched(), MultiRHSBlockCGLinalg< Field >::InnerProductMatrix(), MultiGeneralCoarsenedMatrix< Fobj, CComplex, nbasis >::M(), and MultiRHSBlockCGLinalg< Field >::MulMatrix().

◆ gemmBatched() [2/8]

void GridBLAS::gemmBatched ( int m,
int n,
int k,
ComplexF alpha,
deviceVector< ComplexF * > & Amk,
deviceVector< ComplexF * > & Bkn,
ComplexF beta,
deviceVector< ComplexF * > & Cmn )
inline

Definition at line 151 of file BatchedBlas.h.

References gemmBatched(), and GridBLAS_OP_N.

◆ gemmBatched() [3/8]

void GridBLAS::gemmBatched ( int m,
int n,
int k,
RealD alpha,
deviceVector< RealD * > & Amk,
deviceVector< RealD * > & Bkn,
RealD beta,
deviceVector< RealD * > & Cmn )
inline

Definition at line 166 of file BatchedBlas.h.

References gemmBatched(), and GridBLAS_OP_N.

◆ gemmBatched() [4/8]

void GridBLAS::gemmBatched ( int m,
int n,
int k,
RealF alpha,
deviceVector< RealF * > & Amk,
deviceVector< RealF * > & Bkn,
RealF beta,
deviceVector< RealF * > & Cmn )
inline

Definition at line 181 of file BatchedBlas.h.

References gemmBatched(), and GridBLAS_OP_N.

◆ gemmBatched() [5/8]

void GridBLAS::gemmBatched ( GridBLASOperation_t OpA,
GridBLASOperation_t OpB,
int m,
int n,
int k,
ComplexD alpha,
deviceVector< ComplexD * > & Amk,
deviceVector< ComplexD * > & Bkn,
ComplexD beta,
deviceVector< ComplexD * > & Cmn )
inline

Definition at line 197 of file BatchedBlas.h.

References acceleratorCopyFromDevice(), acceleratorCopyToDevice(), B, conjugate(), GridBLAS_OP_C, GridBLAS_OP_N, GridBLAS_OP_T, gridblasHandle, synchronise(), thread_for, and usecond().

◆ gemmBatched() [6/8]

void GridBLAS::gemmBatched ( GridBLASOperation_t OpA,
GridBLASOperation_t OpB,
int m,
int n,
int k,
ComplexF alpha,
deviceVector< ComplexF * > & Amk,
deviceVector< ComplexF * > & Bkn,
ComplexF beta,
deviceVector< ComplexF * > & Cmn )
inline

Definition at line 444 of file BatchedBlas.h.

References acceleratorCopyToDevice(), GridBLAS_OP_C, GridBLAS_OP_N, GridBLAS_OP_T, gridblasHandle, synchronise(), thread_for, and usecond().

◆ gemmBatched() [7/8]

void GridBLAS::gemmBatched ( GridBLASOperation_t OpA,
GridBLASOperation_t OpB,
int m,
int n,
int k,
RealF alpha,
deviceVector< RealF * > & Amk,
deviceVector< RealF * > & Bkn,
RealF beta,
deviceVector< RealF * > & Cmn )
inline

Definition at line 634 of file BatchedBlas.h.

References acceleratorCopyToDevice(), GridBLAS_OP_C, GridBLAS_OP_N, GridBLAS_OP_T, gridblasHandle, synchronise(), thread_for, and usecond().

◆ gemmBatched() [8/8]

void GridBLAS::gemmBatched ( GridBLASOperation_t OpA,
GridBLASOperation_t OpB,
int m,
int n,
int k,
RealD alpha,
deviceVector< RealD * > & Amk,
deviceVector< RealD * > & Bkn,
RealD beta,
deviceVector< RealD * > & Cmn )
inline

Definition at line 793 of file BatchedBlas.h.

References acceleratorCopyToDevice(), GridBLAS_OP_C, GridBLAS_OP_N, GridBLAS_OP_T, gridblasHandle, synchronise(), thread_for, and usecond().

◆ benchmark()

template<class CComplex>
double GridBLAS::benchmark ( int M,
int N,
int K,
int BATCH )
inline

Definition at line 950 of file BatchedBlas.h.

References acceleratorMemSet(), acceleratorPut(), B, gemmBatched(), K, synchronise(), and usecond().

Field Documentation

◆ gridblasHandle

gridblasHandle_t GridBLAS::gridblasHandle
static

Definition at line 73 of file BatchedBlas.h.

Referenced by gemmBatched(), gemmBatched(), gemmBatched(), gemmBatched(), Init(), NAMESPACE_BEGIN(), and synchronise().

◆ gridblasInit

int GridBLAS::gridblasInit
static

Definition at line 74 of file BatchedBlas.h.

Referenced by Init(), and NAMESPACE_BEGIN().

The documentation for this class was generated from the following files: