Răsfoiți Sursa

fix n_tasks (#2342)

ggml-ci
slaren 2 ani în urmă
părinte
comite
3602ac4255
1 a modificat fișierele cu 3 adăugiri și 0 ștergeri
  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: