fix n_tasks (#2342)

ggml-ci

diff --git a/ggml.c b/ggml.c
index 747a39241b0b6fae1bbca6b84358f529335178f1..9ee4a8d7f687bae9c572e58f724636bfa31cbf28 100644 (file)
--- 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: