Update CMakeLists.txt

YellowRoseCx · Apr 16, 2024 · 0618435 · 0618435
1 parent f49e382
commit 0618435
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -158,6 +158,7 @@ if (LLAMA_HIPBLAS)
         message(STATUS "HIP and hipBLAS found")
         file(GLOB GGML_SOURCES_CUDA "ggml-cuda/*.cu")
         list(APPEND GGML_SOURCES_CUDA "ggml-cuda.cu")
+        list(APPEND GGML_SOURCES_CUDA "ggml.h")
         add_compile_definitions(GGML_USE_HIPBLAS GGML_USE_CUDA SD_USE_CUBLAS)
         add_library(ggml-rocm OBJECT ${GGML_SOURCES_CUDA})
         if (LLAMA_CUDA_FORCE_DMMV)
@@ -453,6 +454,7 @@ endif()
 
 
 
+
 # THIS FILE IS ONLY INTENDED FOR HIPBLAS BUILD PURPOSES ON WINDOWS VISUAL STUDIO.
 # # YOU'RE NOT RECOMMENDED TO USE IT