From 3602ac4255fd4cd589821346290b464a64b955d1 Mon Sep 17 00:00:00 2001 From: slaren Date: Sun, 23 Jul 2023 15:19:39 +0200 Subject: [PATCH] fix n_tasks (#2342) ggml-ci --- ggml.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ggml.c b/ggml.c index 747a39241..9ee4a8d7f 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: