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 }