Skip to content

Commit

Permalink
ggml : fix arm enabled features check (#10961)
Browse files Browse the repository at this point in the history
  • Loading branch information
slaren authored Dec 24, 2024
1 parent 32d6ee6 commit 3327bb0
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions ggml/src/ggml-cpu/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -135,14 +135,20 @@ function(ggml_add_cpu_backend_variant_impl tag_name)
endif()

# show enabled features
if (CMAKE_HOST_SYSTEM_NAME STREQUAL "Windows")
set(FEAT_INPUT_FILE "NUL")
else()
set(FEAT_INPUT_FILE "/dev/null")
endif()

execute_process(
COMMAND ${CMAKE_C_COMPILER} ${ARCH_FLAGS} -dM -E -
INPUT_FILE "/dev/null"
INPUT_FILE ${FEAT_INPUT_FILE}
OUTPUT_VARIABLE ARM_FEATURE
RESULT_VARIABLE ARM_FEATURE_RESULT
)
if (ARM_FEATURE_RESULT)
message(FATAL_ERROR "Failed to get ARM features")
message(WARNING "Failed to get ARM features")
else()
foreach(feature DOTPROD SVE MATMUL_INT8 FMA FP16_VECTOR_ARITHMETIC)
string(FIND "${ARM_FEATURE}" "__ARM_FEATURE_${feature} 1" feature_pos)
Expand Down

0 comments on commit 3327bb0

Please sign in to comment.