Spaces:
Sleeping
Sleeping
Diego Devesa
commited on
Commit
·
06cddad
1
Parent(s):
38e6172
ggml : fix arm enabled features check (llama/10961)
Browse files
ggml/src/ggml-cpu/CMakeLists.txt
CHANGED
|
@@ -135,14 +135,20 @@ function(ggml_add_cpu_backend_variant_impl tag_name)
|
|
| 135 |
endif()
|
| 136 |
|
| 137 |
# show enabled features
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 138 |
execute_process(
|
| 139 |
COMMAND ${CMAKE_C_COMPILER} ${ARCH_FLAGS} -dM -E -
|
| 140 |
-
INPUT_FILE
|
| 141 |
OUTPUT_VARIABLE ARM_FEATURE
|
| 142 |
RESULT_VARIABLE ARM_FEATURE_RESULT
|
| 143 |
)
|
| 144 |
if (ARM_FEATURE_RESULT)
|
| 145 |
-
message(
|
| 146 |
else()
|
| 147 |
foreach(feature DOTPROD SVE MATMUL_INT8 FMA FP16_VECTOR_ARITHMETIC)
|
| 148 |
string(FIND "${ARM_FEATURE}" "__ARM_FEATURE_${feature} 1" feature_pos)
|
|
|
|
| 135 |
endif()
|
| 136 |
|
| 137 |
# show enabled features
|
| 138 |
+
if (CMAKE_HOST_SYSTEM_NAME STREQUAL "Windows")
|
| 139 |
+
set(FEAT_INPUT_FILE "NUL")
|
| 140 |
+
else()
|
| 141 |
+
set(FEAT_INPUT_FILE "/dev/null")
|
| 142 |
+
endif()
|
| 143 |
+
|
| 144 |
execute_process(
|
| 145 |
COMMAND ${CMAKE_C_COMPILER} ${ARCH_FLAGS} -dM -E -
|
| 146 |
+
INPUT_FILE ${FEAT_INPUT_FILE}
|
| 147 |
OUTPUT_VARIABLE ARM_FEATURE
|
| 148 |
RESULT_VARIABLE ARM_FEATURE_RESULT
|
| 149 |
)
|
| 150 |
if (ARM_FEATURE_RESULT)
|
| 151 |
+
message(WARNING "Failed to get ARM features")
|
| 152 |
else()
|
| 153 |
foreach(feature DOTPROD SVE MATMUL_INT8 FMA FP16_VECTOR_ARITHMETIC)
|
| 154 |
string(FIND "${ARM_FEATURE}" "__ARM_FEATURE_${feature} 1" feature_pos)
|