From: slaren <redacted>
Date: Sun, 23 Jul 2023 13:19:39 +0000 (+0200)
Subject: fix n_tasks (#2342)
X-Git-Tag: gguf-v0.4.0~416
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=3602ac4255fd4cd589821346290b464a64b955d1;p=pkg%2Fggml%2Fsources%2Fllama.cpp

fix n_tasks (#2342)

ggml-ci
---

diff --git a/ggml.c b/ggml.c
index 747a3924..9ee4a8d7 100644
--- a/ggml.c
+++ b/ggml.c
@@ -16278,6 +16278,9 @@ struct ggml_cplan ggml_graph_plan(struct ggml_cgraph * cgraph, int n_threads) {
             case GGML_OP_GET_ROWS:
             case GGML_OP_GET_ROWS_BACK:
             case GGML_OP_DIAG:
+                {
+                    n_tasks = 1;
+                } break;
             case GGML_OP_DIAG_MASK_ZERO:
             case GGML_OP_DIAG_MASK_INF:
             case GGML_OP_SOFT_MAX: