root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-11-15 07:19:53 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

only use AMX on x86

Browse Source
This commit is contained in:
slaren 2024-11-14 00:33:16 +01:00
parent e503ad101d
commit fc66c4bf6d
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ggml/src/ggml-amx/CMakeLists.txt
View File

@ -1,4 +1,7 @@
if (CMAKE_COMPILER_IS_GNUCC AND CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 11.0) if (CMAKE_OSX_ARCHITECTURES STREQUAL "x86_64" OR CMAKE_GENERATOR_PLATFORM_LWR MATCHES "^(x86_64|i686|amd64|x64|win32)$" OR
(NOT CMAKE_OSX_ARCHITECTURES AND NOT CMAKE_GENERATOR_PLATFORM_LWR AND
CMAKE_SYSTEM_PROCESSOR MATCHES "^(x86_64|i686|AMD64)$") AND
CMAKE_COMPILER_IS_GNUCC AND CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 11.0)
message(STATUS "Using AMX") message(STATUS "Using AMX")
file(GLOB GGML_HEADERS_AMX "*.h") file(GLOB GGML_HEADERS_AMX "*.h")
@ -100,5 +103,5 @@ if (CMAKE_COMPILER_IS_GNUCC AND CMAKE_CXX_COMPILER_VERSION VERSION_GREATER 11.0)
target_compile_options(ggml-amx PRIVATE ${ARCH_FLAGS}) target_compile_options(ggml-amx PRIVATE ${ARCH_FLAGS})
else() else()
set(GGML_AMX OFF PARENT_SCOPE) set(GGML_AMX OFF PARENT_SCOPE)
message(WARNING "AMX requires gcc version > 11.0. Turning off GGML_AMX.") message(WARNING "AMX requires x86 and gcc version > 11.0. Turning off GGML_AMX.")
endif() endif()