Skip to content

Commit 22985a4

Browse files
committed
musa: apply mublas API changes
Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>
1 parent bdca383 commit 22985a4

File tree

1 file changed

+2
-2
lines changed
  • ggml/src/ggml-cuda/vendors

1 file changed

+2
-2
lines changed

ggml/src/ggml-cuda/vendors/musa.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
#define CUBLAS_OP_N MUBLAS_OP_N
1414
#define CUBLAS_OP_T MUBLAS_OP_T
1515
#define CUBLAS_STATUS_SUCCESS MUBLAS_STATUS_SUCCESS
16-
#define CUBLAS_TF32_TENSOR_OP_MATH MUBLAS_MATH_MODE_DEFAULT
16+
#define CUBLAS_TF32_TENSOR_OP_MATH MUBLAS_TENSOR_OP_MATH
1717
#define CUDA_R_16F MUSA_R_16F
1818
#define CUDA_R_16BF MUSA_R_16BF
1919
#define CUDA_R_32F MUSA_R_32F
@@ -29,7 +29,7 @@
2929
#define cublasSgemm mublasSgemm
3030
#define cublasStatus_t mublasStatus_t
3131
#define cublasOperation_t mublasOperation_t
32-
#define cublasGetStatusString mublasStatus_to_string
32+
#define cublasGetStatusString mublasGetStatusString
3333
#define cudaDataType_t musaDataType_t
3434
#define cudaDeviceCanAccessPeer musaDeviceCanAccessPeer
3535
#define cudaDeviceDisablePeerAccess musaDeviceDisablePeerAccess

0 commit comments

Comments
 (0)