From c1d255e91bb5d95c729319b751ab9503da336ec7 Mon Sep 17 00:00:00 2001 From: matteo serva Date: Tue, 16 Jul 2024 19:39:17 +0200 Subject: [PATCH] Fixed compilation error when using hipblas --- ggml/src/ggml-cuda.cu | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/ggml/src/ggml-cuda.cu b/ggml/src/ggml-cuda.cu index 0d6a9c8f9..b5697015d 100644 --- a/ggml/src/ggml-cuda.cu +++ b/ggml/src/ggml-cuda.cu @@ -130,6 +130,8 @@ static cudaError_t ggml_cuda_device_malloc(void ** ptr, size_t size, int device) } return res; #else + +#if !defined(GGML_USE_HIPBLAS) cudaError_t err; if (getenv("GGML_CUDA_ENABLE_UNIFIED_MEMORY") != nullptr) { @@ -140,6 +142,10 @@ static cudaError_t ggml_cuda_device_malloc(void ** ptr, size_t size, int device) err = cudaMalloc(ptr, size); } return err; +#else + return cudaMalloc(ptr, size); +#endif // !defined(GGML_USE_HIPBLAS) + #endif }