fix n_tasks (#2342)

ggml-ci
2023-07-23 15:19:39 +02:00 · 2023-07-23 15:19:39 +02:00 · 3602ac4255
parent 95a6c595e7
commit 3602ac4255
1 changed files with 3 additions and 0 deletions
--- 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: