From bbe0c5fbaa1142b3b1cd20da92ad31b9b02810f9 Mon Sep 17 00:00:00 2001 From: laurent Date: Fri, 30 Jun 2023 18:47:22 +0100 Subject: [PATCH] Do not use rayon for a single thread (bis). --- candle-core/src/cpu_backend.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/candle-core/src/cpu_backend.rs b/candle-core/src/cpu_backend.rs index 5af7c019..d9141f0f 100644 --- a/candle-core/src/cpu_backend.rs +++ b/candle-core/src/cpu_backend.rs @@ -251,7 +251,7 @@ impl Map2 for MatMul { let dst_p = &mut dst[step * c_skip..]; let num_threads = crate::utils::get_num_threads(); let parallelism = if num_threads > 1 { - Parallelism::Rayon(crate::utils::get_num_threads()) + Parallelism::Rayon(num_threads) } else { Parallelism::None };