Skip to content

Commit 2307659

Browse files
committed
ggml-cpu: add __ARM_FEATURE_DOTPROD guard for gemv/gemm_iq4_nl_4x4_q8_0
1 parent f56013d commit 2307659

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml/src/ggml-cpu/ggml-cpu-aarch64.c

+2-2
Original file line numberDiff line numberDiff line change
@@ -1017,7 +1017,7 @@ void ggml_gemv_iq4_nl_4x4_q8_0(int n, float * restrict s, size_t bs, const void
10171017
UNUSED(ncols_interleaved);
10181018
UNUSED(blocklen);
10191019

1020-
#if ! ((defined(_MSC_VER)) && ! defined(__clang__)) && defined(__aarch64__) && defined(__ARM_NEON)
1020+
#if ! ((defined(_MSC_VER)) && ! defined(__clang__)) && defined(__aarch64__) && defined(__ARM_NEON) && defined(__ARM_FEATURE_DOTPROD)
10211021
if (ggml_cpu_has_neon()) {
10221022
const int8x16_t kvalues = vld1q_s8(kvalues_iq4nl);
10231023
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
35043504
UNUSED(ncols_interleaved);
35053505
UNUSED(blocklen);
35063506

3507-
#if ! ((defined(_MSC_VER)) && ! defined(__clang__)) && defined(__aarch64__) && defined(__ARM_NEON)
3507+
#if ! ((defined(_MSC_VER)) && ! defined(__clang__)) && defined(__aarch64__) && defined(__ARM_NEON) && defined(__ARM_FEATURE_DOTPROD)
35083508
if (ggml_cpu_has_neon()) {
35093509
const int8x16_t kvalues = vld1q_s8(kvalues_iq4nl);
35103510

0 commit comments

Comments
 (0)