4

我正在使用人造丝库:

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 将产生多少线程的文档?其中说:

默认情况下,Rayon 使用与可用 CPU 数量相同的线程数。请注意,在启用了超线程的系统上,这等于逻辑内核的数量,而不是物理内核的数量。

如果要更改产生的线程数,可以将环境变量设置RAYON_NUM_THREADS为所需的线程数或使用ThreadPoolBuilder::build_global函数方法。

但是,这些步骤对我来说并不清楚。如何在我的 Windows 10 PC 上执行此操作?

4

2 回答 2

6

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

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

于 2019-12-06T01:19:43.117 回答
0

如果您不想设置全局,可以创建一个名为“create_pool”的函数。此辅助函数从 num_threads 构造一个 Rayon ThreadPool 对象。

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 中调用您的代码。这会将所有 Rayon 函数限制为您设置的 num_threads。

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

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

于 2022-02-27T23:31:40.537 回答