Template Function multiply(cublasHandle_t, const device_matrix<T>&, const device_vector<T>&, device_vector<T>&, MatrixOp, MatrixOp)¶
Defined in File matrix_multiply.h
Function Documentation¶
-
template<typename
T
>
__host__ voidmultiply
(cublasHandle_t handle, const device_matrix<T> &a, const device_vector<T> &b, device_vector<T> &c, MatrixOp transa = MatrixOp::normal, MatrixOp transb = MatrixOp::normal)¶ Performs matrix-vector multiplication where C = A * B. The transpose and hermetian of A and B can be used instead.
- Return
host
- Template Parameters
T
:
- Parameters
handle
:a
:b
:c
:transa
:transb
: