diff --git a/ggml/src/iqk/iqk_gemm_floats.cpp b/ggml/src/iqk/iqk_gemm_floats.cpp index 56ce52a4..4d0eba8d 100644 --- a/ggml/src/iqk/iqk_gemm_floats.cpp +++ b/ggml/src/iqk/iqk_gemm_floats.cpp @@ -519,7 +519,7 @@ void set_mul_mat_bf16_r16(std::array& funcs) { bool iqk_set_kernels_float(int ne00, int typeA, int typeB, std::array& kernels) { if (typeA == GGML_TYPE_BF16) { - if (ne00 % 32) return false; + if (ne00 % 16) return false; switch (typeB) { #ifdef __AVX512BF16__ case GGML_TYPE_BF16: set_mul_mat_bf16(kernels); break;