diff options
Diffstat (limited to 'ggml/include/ggml-cuda.h')
| -rw-r--r-- | ggml/include/ggml-cuda.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ggml/include/ggml-cuda.h b/ggml/include/ggml-cuda.h index d7903c66..71bb6dcf 100644 --- a/ggml/include/ggml-cuda.h +++ b/ggml/include/ggml-cuda.h @@ -6,6 +6,9 @@ #ifdef GGML_USE_HIPBLAS #define GGML_CUDA_NAME "ROCm" #define GGML_CUBLAS_NAME "hipBLAS" +#elif defined(GGML_USE_MUSA) +#define GGML_CUDA_NAME "MUSA" +#define GGML_CUBLAS_NAME "muBLAS" #else #define GGML_CUDA_NAME "CUDA" #define GGML_CUBLAS_NAME "cuBLAS" |