From 1b90527d78eb4f688077b39a94b03a0b739d2d03 Mon Sep 17 00:00:00 2001 From: slaren Date: Sun, 9 Feb 2025 16:22:56 +0100 Subject: [PATCH] disable for arm --- ggml/src/ggml-cpu/ggml-cpu.c | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ggml/src/ggml-cpu/ggml-cpu.c b/ggml/src/ggml-cpu/ggml-cpu.c index bee1772f4..d3bd0a3e7 100644 --- a/ggml/src/ggml-cpu/ggml-cpu.c +++ b/ggml/src/ggml-cpu/ggml-cpu.c @@ -7832,11 +7832,16 @@ static void ggml_compute_forward_mul_mat_id( const int64_t nr0 = ne01; const int64_t nr1 = cne1; - //int chunk_size = (nr0 + nr1) / nth; + +#if defined(__aarch64__) + // disable for ARM + int chunk_size = (nr0 + nr1) / nth; +#else int chunk_size = 16; if (nr0 == 1 || nr1 == 1) { chunk_size = 64; } +#endif // defined(__aarch64__) int64_t nchunk0 = (nr0 + chunk_size - 1) / chunk_size; int64_t nchunk1 = (nr1 + chunk_size - 1) / chunk_size;