This commit is contained in:
jon-chuang 2023-04-26 22:58:20 +08:00
parent 5bb5327833
commit 8ead56c03a

2
ggml.c
View file

@ -7755,7 +7755,7 @@ static void ggml_compute_forward_mul_mat_f32(
assert(ne3 == ne03);
#if defined(__AVX2__) || defined(__AVX__)
if (ggml_cpu_has_avx2() && ne00 <= 48 || ne00 <= 32) {
if ((ggml_cpu_has_avx2() && ne00 <= 48) || ne00 <= 32) {
// Handle tall and skinny matrices
// TODO(jon-chuang): Also check that we only handle 2D matrices?
assert(ne00 == ne10);