Home Explore Help
Sign In
cturan
/
llama.cpp
mirror of https://github.com/cturan/llama.cpp
1
0
Fork 0
Files Issues 0 Wiki
Browse Source

fix n_tasks (#2342)

ggml-ci
slaren 2 years ago
parent
95a6c595e7
commit
3602ac4255
1 changed files with 3 additions and 0 deletions
Split View Show Diff Stats
  1. 3 0
      ggml.c

+ 3 - 0
ggml.c
View File 

@@ -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: