ggml-cpu: fix macro declaration

Signed-off-by: Aaron Teo <aaron.teo1@ibm.com>
This commit is contained in:
Aaron Teo
2025-06-21 19:08:48 +08:00
parent 3004a79f4b
commit 1cacdd9a36
2 changed files with 3 additions and 2 deletions

View File

@ -467,12 +467,13 @@ function(ggml_add_cpu_backend_variant_impl tag_name)
if (GGML_VXE)
message(STATUS "VX/VXE/VXE2 enabled")
list(APPEND ARCH_FLAGS -mvx -mzvector)
list(APPEND ARCH_DEFINITIONS GGML_NNPA __NNPA__)
list(APPEND ARCH_DEFINITIONS GGML_VXE)
endif()
if (GGML_NNPA)
target_compile_definitions(${GGML_CPU_NAME} PRIVATE GGML_NNPA)
message(STATUS "NNPA enabled")
list(APPEND ARCH_DEFINITIONS GGML_NNPA)
endif()
elseif (CMAKE_SYSTEM_PROCESSOR MATCHES "wasm")
message(STATUS "Wasm detected")

View File

@ -28,7 +28,7 @@
#include <immintrin.h>
#endif
#if defined(__s390x__) && defined(__VEC__)
#if defined(GGML_VXE)
#include <vecintrin.h>
#include <ggml-cpu/ggml-cpu-impl.h>