Update HPLAI_blas.cc

wu-kan · Apr 23, 2021 · 62aafcf · 62aafcf
1 parent 672d498
commit 62aafcf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/blas/HPLAI_blas.cc b/src/blas/HPLAI_blas.cc
@@ -447,8 +447,8 @@ void blas::gemm<HPLAI_T_AFLOAT, HPLAI_T_AFLOAT, HPLAI_T_AFLOAT>(
     HPLAI_T_AFLOAT rone = HPLAI_rone;
     cublasGemmEx(
         HPLAI_DEVICE_BLASPP_QUEUE->handle(),
-        blas::device_trans_const(TRANSA),
-        blas::device_trans_const(TRANSB),
+        blas::device::op2cublas(TRANSA),
+        blas::device::op2cublas(TRANSB),
         M1,
         N1,
         K1,