ggml : fix n_threads_cur initialization with one thread (ggerganov#9538)

* ggml : fix n_threads_cur initialization with one thread * Update ggml/src/ggml.c --------- Co-authored-by: Max Krasnyansky <[email protected]>
NeoZhangJianyu · Sep 18, 2024 · 64c6af3 · 64c6af3
1 parent 0d2f22e
commit 64c6af3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ggml/src/ggml.c b/ggml/src/ggml.c
@@ -20239,6 +20239,7 @@ enum ggml_status ggml_graph_compute(struct ggml_cgraph * cgraph, struct ggml_cpl
             ggml_graph_compute_thread(&threadpool->workers[omp_get_thread_num()]);
         }
     } else {
+        atomic_store_explicit(&threadpool->n_threads_cur, 1, memory_order_relaxed);
         ggml_graph_compute_thread(&threadpool->workers[0]);
     }
 #else