Public Types |
typedef void(* | gemm_func )(const enum CBLAS_ORDER, const enum CBLAS_TRANSPOSE, const enum CBLAS_TRANSPOSE, const int, const int, const int, const T, const T *, const int, const T *, const int, const T, T *, const int, cudaStream_t) |
typedef void(* | __rs )(const CUevent) |
typedef bool(* | __qu )(const CUevent) |
Static Public Member Functions |
static void | __gemm (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 T alpha, const T *A, const int lda, const T *B, const int ldb, const T beta, T *C, const int ldc, cudaStream_t stream) |
static void | __recordDef (const CUevent event) |
static bool | __queryDef (const CUevent event) |
static void | __synchronizeDef (const CUevent event) |
static void | __recordCuda (const CUevent event) |
static bool | __queryCuda (const CUevent event) |
static void | __synchronizeCuda (const CUevent event) |
Data Fields |
cudaStream_t | stream |
gemm_func | func |
__rs | record |
__rs | synchronize |
__qu | query |
Static Public Attributes |
static CUDAStreamSyncTokenPtr | __syncTok |
template<typename T>
struct lama::LUSolver::lama_gemm< T >
Member Typedef Documentation
template<typename T>
typedef void(* lama::LUSolver::lama_gemm< T >::gemm_func)(const enum CBLAS_ORDER, const enum CBLAS_TRANSPOSE, const enum CBLAS_TRANSPOSE, const int, const int, const int, const T, const T *, const int, const T *, const int, const T, T *, const int, cudaStream_t) |
Member Function Documentation
template<typename T >
void lama::LUSolver::lama_gemm< T >::__gemm |
( |
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 T |
alpha, |
|
|
const T * |
A, |
|
|
const int |
lda, |
|
|
const T * |
B, |
|
|
const int |
ldb, |
|
|
const T |
beta, |
|
|
T * |
C, |
|
|
const int |
ldc, |
|
|
cudaStream_t |
stream |
|
) |
| [static] |
Field Documentation
The documentation for this struct was generated from the following file: