Template Struct TypedGemm< float, float >

• Defined in File prod.cpp

Struct Documentation

template<>
struct TypedGemm<float, float>

Public Static Functions

static void gemm(cublasHandle_t handle, CudaCompute computeCapability, cublasOperation_t transa, cublasOperation_t transb, int m, int n, int k, const float *alpha, const float *A, int lda, const float *B, int ldb, const float *beta, float *C, int ldc)

static void batchedGemm(cublasHandle_t handle, CudaCompute computeCapability, cublasOperation_t transa, cublasOperation_t transb, int m, int n, int k, const float *alpha, const float *Aarray[], int lda, const float *Barray[], int ldb, const float *beta, float *Carray[], int ldc, int batchCount)