Function icrar::cuda::mat_mul(cublasLtHandle_t, MatrixOp, MatrixOp, const size_t, const size_t, const size_t, const float *, const float *, float *)
Defined in File matrix_multiply.h
Function Documentation
-
__host__ void icrar::cuda::mat_mul(cublasLtHandle_t handle, MatrixOp transa, MatrixOp transb, const size_t m, const size_t n, const size_t k, const float *a, const float *b, float *out)