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

fix ColMajor and RowMajor replacement

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