Selaa lähdekoodia

ggml : fix #if for f32_f32 mul_mat (CLBlast) (#1229)

Georgi Gerganov 2 vuotta sitten
vanhempi
sitoutus
ec728e44d7
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      ggml.c

+ 1 - 1
ggml.c

@@ -11592,7 +11592,7 @@ void ggml_graph_compute(struct ggml_context * ctx, struct ggml_cgraph * cgraph)
 #endif
                         } else if (node->src0->type == GGML_TYPE_F32 && node->src1->type == GGML_TYPE_F32) {
                             cur = 0;
-#if defined(GGML_USE_ACCELERATE) || defined(GGML_USE_OPENBLAS) || defined(GGML_USE_CUBLAS)
+#if defined(GGML_USE_ACCELERATE) || defined(GGML_USE_OPENBLAS) || defined(GGML_USE_CUBLAS) || defined(GGML_USE_CLBLAST)
                             if (ggml_compute_forward_mul_mat_use_blas(node->src0, node->src1, node)) {
                                 node->n_tasks = 1;
                             }