| 194 | */ |
| 195 | template <typename _Scalar> |
| 196 | void cublasGemm( |
| 197 | cublasOperation_t transA, cublasOperation_t transB, |
| 198 | int m, int n, int k, |
| 199 | const _Scalar* alpha, |
| 200 | const _Scalar* A, int lda, |
| 201 | const _Scalar* B, int ldb, |
| 202 | const _Scalar* beta, |
| 203 | _Scalar* C, int ldc) |
| 204 | { |
| 205 | cublasgemmImpl(transA, transB, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc); |
| 206 | } |
| 207 | |
| 208 | private: |
| 209 | #define CUBLAS_GEMM3M_FACTORY(scalar, op) \ |