Skip to content

Commit 4ad6efa

Browse files
committed
ggml-cpu: diagnose why __NNPA__ macro is not being defined
Signed-off-by: Aaron Teo <[email protected]>
1 parent 0e571dd commit 4ad6efa

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

ggml/src/CMakeLists.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,10 @@ if (GGML_ALL_WARNINGS)
5555
endif()
5656
endif()
5757

58+
if (GGML_NNPA)
59+
add_compile_definitions(__NNPA__)
60+
endif()
61+
5862
if (GGML_LTO)
5963
include(CheckIPOSupported)
6064
check_ipo_supported(RESULT result OUTPUT output)

ggml/src/ggml-cpu/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -467,6 +467,7 @@ function(ggml_add_cpu_backend_variant_impl tag_name)
467467
if (GGML_VXE)
468468
message(STATUS "VX/VXE/VXE2 enabled")
469469
list(APPEND ARCH_FLAGS -mvx -mzvector)
470+
list(APPEND ARCH_DEFINITIONS GGML_NNPA __NNPA__)
470471
endif()
471472

472473
if (GGML_NNPA)

0 commit comments

Comments
 (0)