From 290f81aa4fea75390a18edf33fb64c0011347946 Mon Sep 17 00:00:00 2001 From: slaren Date: Sun, 24 Mar 2024 16:34:01 +0100 Subject: [PATCH] update cmake for HIP --- CMakeLists.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index c715aaf7c..030b3c4f3 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -521,7 +521,9 @@ if (LLAMA_HIPBLAS) message(STATUS "HIP and hipBLAS found") set(GGML_HEADERS_ROCM ggml-cuda.h) - set(GGML_SOURCES_ROCM ggml-cuda.cu) + + file(GLOB GGML_SOURCES_ROCM "ggml-cuda/*.cu") + list(APPEND GGML_SOURCES_ROCM "ggml-cuda.cu") add_compile_definitions(GGML_USE_HIPBLAS GGML_USE_CUBLAS)