ggml-cpu: fix macro declaration

Signed-off-by: Aaron Teo <[email protected]>

Author: taronaeo

ggml/src/ggml-cpu/CMakeLists.txt

@@ -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")

ggml/src/ggml-impl.h

@@ -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>