Преглед на файлове

ggml : arm repack fix build

Georgi Gerganov преди 1 месец
родител
ревизия
a63cbafbbc
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      ggml/src/ggml-cpu/arch/arm/repack.cpp

+ 1 - 1
ggml/src/ggml-cpu/arch/arm/repack.cpp

@@ -24,7 +24,7 @@
 
 
 #define UNUSED GGML_UNUSED
 #define UNUSED GGML_UNUSED
 
 
-#if defined(__aarch64__) && defined(__ARM_NEON) && defined(__ARM_FEATURE_MATMUL_INT8)
+#if defined(__aarch64__) && defined(__ARM_NEON) && (defined(__ARM_FEATURE_MATMUL_INT8) || defined(__ARM_FEATURE_DOTPROD))
 static inline void decode_q4_Kx8_scales_mins(const uint8_t * scales_in,
 static inline void decode_q4_Kx8_scales_mins(const uint8_t * scales_in,
                                              int16x8_t *     out_mins,
                                              int16x8_t *     out_mins,
                                              int8_t *        out_scales) {
                                              int8_t *        out_scales) {