1
0
slaren 2 жил өмнө
parent
commit
3602ac4255
1 өөрчлөгдсөн 3 нэмэгдсэн , 0 устгасан
  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: