Skip to content

Commit 5be2497

Browse files
committed
Fix pointer
1 parent 7a497a5 commit 5be2497

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

source/module_base/blas_connector.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ void BlasConnector::gemm(const char transa, const char transb, const int m, cons
202202
#ifdef __CUDA
203203
cublasOperation_t cutransA = judge_trans_op(false, transa, "gemm_op");
204204
cublasOperation_t cutransB = judge_trans_op(false, transb, "gemm_op");
205-
cublasErrcheck(cublasSgemm(cublas_handle, cutransA, cutransB, m, n, k, (float2*)&alpha, a, lda, b, ldb, (float2*)&beta, c, ldc));
205+
cublasErrcheck(cublasSgemm(cublas_handle, cutransA, cutransB, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc));
206206
#endif
207207
}
208208
}
@@ -227,7 +227,7 @@ void BlasConnector::gemm(const char transa, const char transb, const int m, cons
227227
#ifdef __CUDA
228228
cublasOperation_t cutransA = judge_trans_op(false, transa, "gemm_op");
229229
cublasOperation_t cutransB = judge_trans_op(false, transb, "gemm_op");
230-
cublasErrcheck(cublasDgemm(cublas_handle, cutransA, cutransB, m, n, k, (double2*)&alpha, a, lda, b, ldb, (double2*)&beta, c, ldc));
230+
cublasErrcheck(cublasDgemm(cublas_handle, cutransA, cutransB, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc));
231231
#endif
232232
}
233233
}
@@ -252,7 +252,7 @@ void BlasConnector::gemm(const char transa, const char transb, const int m, cons
252252
#ifdef __CUDA
253253
cublasOperation_t cutransA = judge_trans_op(false, transa, "gemm_op");
254254
cublasOperation_t cutransB = judge_trans_op(false, transb, "gemm_op");
255-
cublasErrcheck(cublasCgemm(cublas_handle, cutransA, cutransB, m, n, k, (float2*)&alpha, a, lda, b, ldb, (float2*)&beta, c, ldc));
255+
cublasErrcheck(cublasCgemm(cublas_handle, cutransA, cutransB, m, n, k, (float2*)&alpha, (float2*)a, lda, (float2*)b, ldb, (float2*)&beta, (float2*)c, ldc));
256256
#endif
257257
}
258258
}
@@ -277,7 +277,7 @@ void BlasConnector::gemm(const char transa, const char transb, const int m, cons
277277
#ifdef __CUDA
278278
cublasOperation_t cutransA = judge_trans_op(false, transa, "gemm_op");
279279
cublasOperation_t cutransB = judge_trans_op(false, transb, "gemm_op");
280-
cublasErrcheck(cublasZgemm(cublas_handle, cutransA, cutransB, m, n, k, (double2*)&alpha, a, lda, b, ldb, (double2*)&beta, c, ldc));
280+
cublasErrcheck(cublasZgemm(cublas_handle, cutransA, cutransB, m, n, k, (double2*)&alpha, (double2*)a, lda, (double2*)b, ldb, (double2*)&beta, (double2*)c, ldc));
281281
#endif
282282
}
283283
}
@@ -322,7 +322,7 @@ void BlasConnector::gemv(const char trans, const int m, const int n,
322322
else if (device_type == base_device::AbacusDevice_t::GpuDevice) {
323323
#ifdef __CUDA
324324
cublasOperation_t cutrans = judge_trans_op(false, trans, "gemv_op");
325-
cublasErrcheck(cublasCgemv(cublas_handle, cutrans, m, n, (float2*)&alpha, A, lda, X, incX, (float2*)&beta, Y, incY));
325+
cublasErrcheck(cublasCgemv(cublas_handle, cutrans, m, n, (float2*)&alpha, (float2*)A, lda, (float2*)X, incX, (float2*)&beta, (float2*)Y, incY));
326326
#endif
327327
}
328328
}
@@ -337,7 +337,7 @@ void BlasConnector::gemv(const char trans, const int m, const int n,
337337
else if (device_type == base_device::AbacusDevice_t::GpuDevice) {
338338
#ifdef __CUDA
339339
cublasOperation_t cutrans = judge_trans_op(false, trans, "gemv_op");
340-
cublasErrcheck(cublasZgemv(cublas_handle, cutrans, m, n, (double2*)&alpha, A, lda, X, incX, (double2*)&beta, Y, incY));
340+
cublasErrcheck(cublasZgemv(cublas_handle, cutrans, m, n, (double2*)&alpha, (double2*)A, lda, (double2*)X, incX, (double2*)&beta, (double2*)Y, incY));
341341
#endif
342342
}
343343
}

0 commit comments

Comments
 (0)