想调用CUBLAS的Sgemm函数求矩阵相乘问题:C[hA][wB]=A[hA][wA]*B[hB][wB](wA=hB)
由于矩阵A和B都是在CPU中生成的,按行优先存储,传到d_A和d_B的显存时,仍然按照行优先存储,而CUBLAS矩阵是按列存储的,所以这里是不是应该选择’t’,‘t’?执行以下函数:
cublasSgemm(‘t’,‘t’,wA,hB,hA,1.0,d_A,wA,d_B,wB,0.0,d_C,wB);
发现当hA=wA=hB=wB时,计算结果是正确的,得到的d_C是实际所求C矩阵的转置。
但是当hA!=wA!=wB时,就会报错:On entry to SGEMM parameter number 13 had an illegal value,大家能说说是什么原因么?
是不是我矩阵来回转置出现错误了?谢谢大家了