提交 de856da9 编写于 作者: T tensor-tang

fix ColMajor and RowMajor replacement

上级 00ee6c3c
......@@ -178,7 +178,7 @@ void Blas<platform::CPUDeviceContext>::GEMM(CBLAS_TRANSPOSE transA,
const int lda = M;
const int ldb = K;
const int ldc = M;
CBlas<T>::SMM_GEMM(&transa, &transb, &M, &N, &K, &alpha, A, &lda, B, &ldb,
CBlas<T>::SMM_GEMM(&transa, &transb, &N, &M, &K, &alpha, B, &ldb, A, &lda,
&beta, C, &ldc);
} else {
#endif
......
......@@ -82,8 +82,8 @@ void MklSmmCompare(int m, int n, int k) {
const int lda = m;
const int ldb = k;
const int ldc = m;
paddle::operators::math::CBlas<T>::SMM_GEMM(&transa, &transb, &m, &n, &k,
&alpha, A, &lda, B, &ldb, &beta,
paddle::operators::math::CBlas<T>::SMM_GEMM(&transa, &transb, &n, &m, &k,
&alpha, B, &ldb, A, &lda, &beta,
CSMM, &ldc);
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册