|
| 1 | +diff --git a/ggml/src/ggml-cpu/CMakeLists.txt b/ggml/src/ggml-cpu/CMakeLists.txt |
| 2 | +index 66a5ad8d..f71c7dec 100644 |
| 3 | +--- a/ggml/src/ggml-cpu/CMakeLists.txt |
| 4 | ++++ b/ggml/src/ggml-cpu/CMakeLists.txt |
| 5 | +@@ -51,7 +51,7 @@ function(ggml_add_cpu_backend_variant_impl tag_name) |
| 6 | + |
| 7 | + target_compile_features(${GGML_CPU_NAME} PRIVATE c_std_11 cxx_std_17) |
| 8 | + target_include_directories(${GGML_CPU_NAME} PRIVATE . ggml-cpu) |
| 9 | +- |
| 10 | ++ list(APPEND ARCH_FLAGS -fzvector -m64 -march=z15) |
| 11 | + if (APPLE AND GGML_ACCELERATE) |
| 12 | + find_library(ACCELERATE_FRAMEWORK Accelerate) |
| 13 | + if (ACCELERATE_FRAMEWORK) |
| 14 | +@@ -94,7 +94,7 @@ function(ggml_add_cpu_backend_variant_impl tag_name) |
| 15 | + target_compile_definitions(${GGML_CPU_NAME} PRIVATE GGML_USE_CPU_HBM) |
| 16 | + |
| 17 | + target_link_libraries(${GGML_CPU_NAME} PUBLIC memkind) |
| 18 | +- endif() |
| 19 | ++ endif() |
| 20 | + |
| 21 | + if (GGML_SYSTEM_ARCH STREQUAL "ARM") |
| 22 | + message(STATUS "ARM detected") |
| 23 | +@@ -463,11 +463,10 @@ function(ggml_add_cpu_backend_variant_impl tag_name) |
| 24 | + message(WARNING "Unknown target. If you are compiling for z14 and earlier, you might have to add -DGGML_VXE=OFF.") |
| 25 | + list(APPEND ARCH_FLAGS -march=native -mtune=native) |
| 26 | + endif() |
| 27 | +- |
| 28 | + if (GGML_VXE) |
| 29 | + message(STATUS "VX/VXE/VXE2 enabled") |
| 30 | + list(APPEND ARCH_FLAGS -mvx -mzvector) |
| 31 | +- list(APPEND ARCH_DEFINITIONS GGML_VXE) |
| 32 | ++ list(APPEND ARCH_DEFINITIONS GGML_VXE) |
| 33 | + endif() |
| 34 | + |
| 35 | + if (GGML_NNPA) |
| 36 | +@@ -480,6 +479,17 @@ function(ggml_add_cpu_backend_variant_impl tag_name) |
| 37 | + else() |
| 38 | + message(WARNING "Unknown CPU architecture. Falling back to generic implementations.") |
| 39 | + list(APPEND ARCH_FLAGS -DGGML_CPU_GENERIC) |
| 40 | ++ list(APPEND ARCH_DEFINITIONS GGML_VXE) |
| 41 | ++ message(STATUS "Added GGML_VXE Flag") |
| 42 | ++ if(NOT DEFINED TARGET_ARCH) |
| 43 | ++ set(TARGET_ARCH 13) |
| 44 | ++ endif() |
| 45 | ++ if(TARGET_ARCH GREATER 13) |
| 46 | ++ target_compile_options(${GGML_CPU_NAME} PRIVATE "-qarch=${TARGET_ARCH}") |
| 47 | ++ endif() |
| 48 | ++ target_include_directories(${GGML_CPU_NAME} PRIVATE /usr/include) |
| 49 | ++ target_link_libraries(${GGML_CPU_NAME} PRIVATE "/usr/lpp/cbclib/lib/libmass.arch${TARGET_ARCH}.a") |
| 50 | ++ message(STATUS "Found MASS: /usr/lpp/cbclib/lib/libmass.arch${TARGET_ARCH}.a") |
| 51 | + endif() |
| 52 | + |
| 53 | + if (GGML_CPU_REPACK) |
0 commit comments