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

slaren · max-krasnyansky · arthw · commit a54e0f2efc43 · 2024-11-15T12:34:22.000+08:00
* ggml : fix n_threads_cur initialization with one thread

* Update ggml/src/ggml.c

---------

Co-authored-by: Max Krasnyansky &lt;quic_maxk@quicinc.com&gt;
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

Original file line number	Diff line number	Diff line change
`@@ -20239,6 +20239,7 @@ enum ggml_status ggml_graph_compute(struct ggml_cgraph * cgraph, struct ggml_cpl`
`20239`	`20239`	`ggml_graph_compute_thread(&threadpool->workers[omp_get_thread_num()]);`
`20240`	`20240`	`}`
`20241`	`20241`	`} else {`
	`20242`	`+ atomic_store_explicit(&threadpool->n_threads_cur, 1, memory_order_relaxed);`
`20242`	`20243`	`ggml_graph_compute_thread(&threadpool->workers[0]);`
`20243`	`20244`	`}`
`20244`	`20245`	`#else`