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

* ggml : fix n_threads_cur initialization with one thread * Update ggml/src/ggml.c --------- Co-authored-by: Max Krasnyansky <quic_maxk@quicinc.com>
2024-09-22 21:16:20 +00:00 · 2024-09-18 19:13:08 +02:00 · 2024-09-18 19:13:08 +02:00 · 64c6af3195
commit 64c6af3195
parent 0d2f22e45c
1 changed files with 1 additions and 0 deletions
--- 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