From da122af024eb302995317e80a0e93035dc32575a Mon Sep 17 00:00:00 2001 From: gwjr <502526+gwjr@users.noreply.github.com> Date: Thu, 16 Nov 2023 18:45:33 +0000 Subject: [PATCH] Add openBLAS support for sgemm() in compute_forward_out_prod() --- ggml.c | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ggml.c b/ggml.c index 8fed22a01396e..f43962acac832 100644 --- a/ggml.c +++ b/ggml.c @@ -9631,9 +9631,9 @@ static void ggml_compute_forward_out_prod_f32( // compute by src0 rows // TODO: #if defined(GGML_USE_CUBLAS) ggml_cuda_out_prod - // TODO: #if defined(GGML_USE_OPENBLAS) || defined(GGML_USE_CLBLAST) + // TODO: #if defined(GGML_USE_CLBLAST) -#if defined(GGML_USE_ACCELERATE) +#if defined(GGML_USE_ACCELERATE) || defined(GGML_USE_OPENBLAS) bool use_blas = ggml_is_matrix(src0) && ggml_is_matrix(src1) && ggml_is_contiguous(src0) && @@ -9641,7 +9641,7 @@ static void ggml_compute_forward_out_prod_f32( #endif if (params->type == GGML_TASK_INIT) { -#if defined(GGML_USE_ACCELERATE) // gemm beta will zero dst +#if defined(GGML_USE_ACCELERATE) || defined(GGML_USE_OPENBLAS) // gemm beta will zero dst if (use_blas) { return; } @@ -9654,7 +9654,7 @@ static void ggml_compute_forward_out_prod_f32( return; } -#if defined(GGML_USE_ACCELERATE) +#if defined(GGML_USE_ACCELERATE) || defined(GGML_USE_OPENBLAS) if (use_blas) { if (params->ith != 0) { // All threads other than the first do no work. return;