update CMakeLists.txt

ggerganov · Sep 18, 2024 · 38b955c · 38b955c
1 parent 9504b9f
commit 38b955c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ggml/src/CMakeLists.txt b/ggml/src/CMakeLists.txt
@@ -269,8 +269,12 @@ if (GGML_AMX)
     message(STATUS "Using AMX")
 
     add_compile_definitions(GGML_USE_AMX)
-    set(GGML_HEADERS_AMX ggml-amx/mmq.h)
-    set(GGML_SOURCES_AMX ggml-amx/mmq.cpp)
+
+    file(GLOB   GGML_HEADERS_AMX "ggml-amx/*.h")
+    list(APPEND GGML_HEADERS_AMX "../include/ggml-amx.h")
+
+    file(GLOB   GGML_SOURCES_AMX "ggml-amx/*.cpp")
+    list(APPEND GGML_SOURCES_AMX "ggml-amx.cpp")
 endif()
 
 if (GGML_CUDA)