Forráskód Böngészése

fix n_tasks (#2342)

ggml-ci
slaren 2 éve
szülő
commit
3602ac4255
1 módosított fájl, 3 hozzáadás és 0 törlés
  1. 3 0
      ggml.c

+ 3 - 0
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: