diff options
author | slaren <slarengh@gmail.com> | 2023-07-23 15:19:39 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-23 15:19:39 +0200 |
commit | 3602ac4255fd4cd589821346290b464a64b955d1 (patch) | |
tree | ed88e0e25f555a25c7ec6cc1cd522f52b971a320 /ggml.c | |
parent | 95a6c595e7ca8dbe47ccf8824e04213e10357f9a (diff) |
fix n_tasks (#2342)
ggml-ci
Diffstat (limited to 'ggml.c')
-rw-r--r-- | ggml.c | 3 |
1 files changed, 3 insertions, 0 deletions
@@ -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: |