如何更改人造丝使用的线程数？

Question

我在用人造丝库：

extern crate rayon;

const N: usize = 1_000_000_000;
const W: f64 = 1f64/(N as f64);

fn f(x: f64) -> f64 {
    4.0/(1.0+x*x)
}

fn main() { 
    use rayon::prelude::*;
    let sum : f64 = (0..N)
        .into_par_iter()
        .map(|i| f(W*((i as f64)+0.5)))
        .sum::<f64>();
    println!("pi = {}", W*sum);
}

我希望使用不同数量的线程来运行这段代码: 1、2、3和4。

我读过关于人造丝会产生多少个线程？的文档，其中说：

默认情况下，Rayon使用的线程数与可用CPU数相同。注意，在启用超线程的系统上，这等于逻辑核的数量，而不是物理核的数量。 如果要更改生成的线程数，可以将环境变量RAYON_NUM_THREADS设置为所需的线程数，或使用函数方法。

然而，这些步骤对我来说并不清楚。我怎样才能在我的Windows 10 PC上做到这一点？

Answer 1

只需在fn main()中包含。num_threads接受线程数。

rayon::ThreadPoolBuilder::new().num_threads(4).build_global().unwrap();

Answer 2

如果不想设置全局函数，则可以创建一个名为“create _pool”的函数。此助手函数从ThreadPool构造一个Rayon num_threads对象。

pub fn create_pool(num_threads: usize) -> Result<rayon::ThreadPool, YOURERRORENUM> {
   match rayon::ThreadPoolBuilder::new()
      .num_threads(num_threads)
      .build()
   {
      Err(e) => Err(e.into()),
      Ok(pool) => Ok(pool),
   }
}

然后从这个create_pool内部调用您的代码。这将限制所有的人造丝功能到您设置的num_threads。

[...]
    create_pool(num_threads)?.install(|| {
       YOURCODE
     })?;
[...]

有关更多信息，请参见https://towardsdatascience.com/nine-rules-for-writing-python-extensions-in-rust-d35ea3a4ec29