Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-08-30 17:10:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

fixed a bug that resulted in the program hanging

Browse Source
This commit is contained in:
Concedo
2024-01-09 18:12:58 +08:00
parent 66533c8424
commit 693f6493e5
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ggml.c
+3 -2
View File
@@ -16557,7 +16557,7 @@ static thread_ret_t ggml_graph_compute_thread(void * data) {
atomic_store(&state->shared->n_active, n_threads);
atomic_store(&state->shared->node_n, node_n);
} else {
// wait for other threads to finish
// wait for other threads to finish
const int last = node_n;
const bool do_yield = last < 0 || cgraph->nodes[last]->op == GGML_OP_MUL_MAT;
@@ -16573,7 +16573,8 @@ static thread_ret_t ggml_graph_compute_thread(void * data) {
}
node_n = atomic_load(&state->shared->node_n);
} while (node_n == last);
if (node_n != last) break;
};
}
// check if we should stop