From 979010cdba8d147c5fcab2e8dfd7e8d3c9782a14 Mon Sep 17 00:00:00 2001 From: jon-chuang Date: Sun, 30 Apr 2023 21:02:55 +0800 Subject: [PATCH] minor --- ggml.c | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ggml.c b/ggml.c index 0a3537723..31592c758 100644 --- a/ggml.c +++ b/ggml.c @@ -8124,9 +8124,8 @@ static void ggml_compute_forward_mul_mat_f32( assert(ne3 == ne03); #if (__AVX512F__ || __AVX2__ || __AVX__) && __FMA__ + // Handle tall and skinny matrices 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? if (params->type == GGML_TASK_INIT || params->type == GGML_TASK_FINALIZE) { return; }