From 38b955cf367f31d6e8d7230f012eed669725b9a5 Mon Sep 17 00:00:00 2001 From: mingfeima Date: Wed, 18 Sep 2024 00:12:02 -0700 Subject: [PATCH] update CMakeLists.txt --- ggml/src/CMakeLists.txt | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/ggml/src/CMakeLists.txt b/ggml/src/CMakeLists.txt index 50e605482045f..f184eba96a426 100644 --- 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)