Skip to content

GitLab

Unverified Commit 9d0f67d6 authored by hclearner's avatar hclearner Committed by GitHub
Browse files

Update cublas_wrapper.h

parent 43eabe75
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
cuda/utils/cublas_wrapper.h
View file @ 9d0f67d6
...@@ -77,7 +77,7 @@ inline cublasStatus_t cublasXgemm(cublasHandle_t handle, ...@@ -77,7 +77,7 @@ inline cublasStatus_t cublasXgemm(cublasHandle_t handle,
const __half *B, int ldb, const __half *B, int ldb,
const __half *beta, const __half *beta,
__half *C, int ldc) { __half *C, int ldc) {
#ifdef MOE_HIP_DIFF #ifdef FMOE_USE_HIP
return rocblas_hgemm(handle, transa, transb, m, n, k, (const rocblas_half*)alpha, (const rocblas_half* )A, lda, (const rocblas_half* )B, ldb, (const rocblas_half*)beta, (rocblas_half* )C, ldc); return rocblas_hgemm(handle, transa, transb, m, n, k, (const rocblas_half*)alpha, (const rocblas_half* )A, lda, (const rocblas_half* )B, ldb, (const rocblas_half*)beta, (rocblas_half* )C, ldc);
#else #else
return cublasHgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc); return cublasHgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc);
...@@ -92,7 +92,7 @@ inline cublasStatus_t cublasXgemm(cublasHandle_t handle, ...@@ -92,7 +92,7 @@ inline cublasStatus_t cublasXgemm(cublasHandle_t handle,
const c10::Half *B, int ldb, const c10::Half *B, int ldb,
const c10::Half *beta, const c10::Half *beta,
c10::Half *C, int ldc) { c10::Half *C, int ldc) {
#ifdef MOE_HIP_DIFF #ifdef FMOE_USE_HIP
return rocblas_hgemm(handle, transa, transb, m, n, k, return rocblas_hgemm(handle, transa, transb, m, n, k,
(const rocblas_half*)alpha, (const rocblas_half*)alpha,
(const rocblas_half*)A, lda, (const rocblas_half*)A, lda,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment