Skip to content

Commit bc8ed09

Browse files
committed
Fix blas usgae
1 parent c778f26 commit bc8ed09

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

source/module_base/test/blas_connector_test.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,7 @@ TEST(blas_connector, Gemv) {
365365
answer_n[j] = alpha_const * c_dot_n[j] + beta_const * result_n[j];
366366
}
367367
BlasConnector::gemv(transa_n, size_m, size_n, alpha_const, a_const.data(), lda,
368-
x_const_n.data(), incx, beta_const, result_m.data(), incy);
368+
x_const_m.data(), incx, beta_const, result_n.data(), incy);
369369

370370
for (int j = 0; j < size_n; j++) {
371371
for (int i = 0; i < size_m; i++) {
@@ -374,7 +374,7 @@ TEST(blas_connector, Gemv) {
374374
answer_c[j] = alpha_const * c_dot_c[j] + beta_const * result_c[j];
375375
}
376376
BlasConnector::gemv(transa_h, size_m, size_n, alpha_const, a_const.data(), lda,
377-
x_const_n.data(), incx, beta_const, result_m.data(), incy);
377+
x_const_c.data(), incx, beta_const, result_c.data(), incy);
378378

379379
for (int i = 0; i < size_m; i++) {
380380
EXPECT_DOUBLE_EQ(answer_m[i].real(), result_m[i].real());
@@ -519,7 +519,7 @@ TEST(blas_connector, Gemm) {
519519
beta_const * result[i + j * ldc];
520520
}
521521
}
522-
BlasConnector::gemm(transa_m, transb_m, size_m, size_n, size_k, alpha_const,
522+
BlasConnector::gemm_cm(transa_m, transb_m, size_m, size_n, size_k, alpha_const,
523523
a_const.data(), lda, b_const.data(), ldb, beta_const,
524524
result.data(), ldc);
525525

0 commit comments

Comments
 (0)