ggml-cpu: add __ARM_FEATURE_DOTPROD guard for gemv/gemm_iq4_nl_4x4_q8_0

ggerganov · Nov 27, 2024 · 2307659 · 2307659
1 parent f56013d
commit 2307659
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ggml/src/ggml-cpu/ggml-cpu-aarch64.c b/ggml/src/ggml-cpu/ggml-cpu-aarch64.c
@@ -1017,7 +1017,7 @@ void ggml_gemv_iq4_nl_4x4_q8_0(int n, float * restrict s, size_t bs, const void
     UNUSED(ncols_interleaved);
     UNUSED(blocklen);
 
-#if ! ((defined(_MSC_VER)) && ! defined(__clang__)) && defined(__aarch64__) && defined(__ARM_NEON)
+#if ! ((defined(_MSC_VER)) && ! defined(__clang__)) && defined(__aarch64__) && defined(__ARM_NEON) && defined(__ARM_FEATURE_DOTPROD)
     if (ggml_cpu_has_neon()) {
         const int8x16_t kvalues = vld1q_s8(kvalues_iq4nl);
         const block_q8_0 * a_ptr = (const block_q8_0 *) vy;
@@ -3504,7 +3504,7 @@ void ggml_gemm_iq4_nl_4x4_q8_0(int n, float * restrict s, size_t bs, const void
     UNUSED(ncols_interleaved);
     UNUSED(blocklen);
 
-#if ! ((defined(_MSC_VER)) && ! defined(__clang__)) && defined(__aarch64__) && defined(__ARM_NEON)
+#if ! ((defined(_MSC_VER)) && ! defined(__clang__)) && defined(__aarch64__) && defined(__ARM_NEON) && defined(__ARM_FEATURE_DOTPROD)
     if (ggml_cpu_has_neon()) {
         const int8x16_t kvalues = vld1q_s8(kvalues_iq4nl);