fix ColMajor and RowMajor replacement

guochaorong-patch-1
tensor-tang 7 years ago
parent 00ee6c3c17
commit de856da9a6

@ -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);
};

Loading…
Cancel
Save