cublasSgemm函数的使用笔记

Posted 2021-12-22 ShaderJoy

实际的运算为C^T=alpha*B^T*A^T+beta*C^T

由于C^T的存储也是按列的，所以计算结果取出来刚好等于 C，可以省去转置的步骤。

注：

maxtrix_size.uiWB 表示，B^T 的行数、C^T的行数

maxtrix_size.uiHA 表示，A^T 的列数

maxtrix_size.uiWA 表示，B^T 的列数、A^T的行数

相关函数和参数说明：

cublasHandle_t 的申请和释放

申请：使用cublasCreate((cublasHandle_t *handle));

释放：使用cublasDestroy_v2 (cublasHandle_t handle);

cublasOperation_t　可能的取值：

CUBLAS_OP_N：不转置（但是由于按列存储，其实传进去的矩阵还是被转置了）

CUBLAS_OP_T：转置

CUBLAS_OP_C：共轭转置操作

​