Skip to content

Commit 1b90527

Browse files
committed
disable for arm
1 parent 0f0d8c3 commit 1b90527

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

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

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7832,11 +7832,16 @@ static void ggml_compute_forward_mul_mat_id(
78327832
const int64_t nr0 = ne01;
78337833
const int64_t nr1 = cne1;
78347834

7835-
//int chunk_size = (nr0 + nr1) / nth;
7835+
7836+
#if defined(__aarch64__)
7837+
// disable for ARM
7838+
int chunk_size = (nr0 + nr1) / nth;
7839+
#else
78367840
int chunk_size = 16;
78377841
if (nr0 == 1 || nr1 == 1) {
78387842
chunk_size = 64;
78397843
}
7844+
#endif // defined(__aarch64__)
78407845

78417846
int64_t nchunk0 = (nr0 + chunk_size - 1) / chunk_size;
78427847
int64_t nchunk1 = (nr1 + chunk_size - 1) / chunk_size;

0 commit comments

Comments
 (0)