cublasSgemm函数

最新推荐文章于 2025-05-22 22:59:05 发布

小_小_杨_

最新推荐文章于 2025-05-22 22:59:05 发布

阅读量5.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： cuda

本文链接：https://blog.youkuaiyun.com/u012235274/article/details/53677382

cuda 专栏收录该内容

8 篇文章

订阅专栏

本文详细介绍了cuBLAS库中sgemm函数的使用方法，包括参数含义、矩阵存储方式及运算规则，并提供了如何创建和销毁cuBLAS句柄的具体步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

cublasStatus_t cublasSgemm (
cublasHandle_t handle, 
cublasOperation_t transa,
cublasOperation_t transb, 
int m,
int n,
int k,
const float *alpha, /* host or device pointer */  
const float *A, 
int lda,
const float *B,
int ldb, 
const float *beta, /* host or device pointer */  
float *C,
int ldc);

在cublas里面所有矩阵都是使用列优先进行存储的，因此lda，ldb，ldc表示的是矩阵的行数。
矩阵OP(Ａ)的维度是ｍ×ｋ
矩阵OP(B)的维度是ｋ×ｎ
矩阵C的维度是ｍ×ｎ
运算为C=alpha*A*B+beta*C
cublasHandle_t使用函数cublasCreate((cublasHandle_t *handle));进行申请，之后使用cublasDestroy_v2 (cublasHandle_t handle)进行释放。
cublasOperation_t　的值为CUBLAS_OP_N和CUBLAS_OP_T分别表示不转置和转置。