2018年8月12日
cuda中用cublas库做矩阵乘法
摘要： 这里矩阵C=A*B，原始文档给的公式是C=alpha*A*B+beta*C，所以这里alpha=1，beta=0。 主要使用cublasSgemm这个函数，这个函数的第二个参数有三种类型，这里CUBLAS_OP_N求出来矩阵结果是按行排列，所以不需要转置了。 如果用CUBLAS_OP_T参数求得的结 阅读全文
posted @ 2018-08-12 16:47 Dsp Tian 阅读(4466) 评论(0) 推荐(0) 编辑