slaren 2 лет назад
Родитель
Сommit
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: