blob: 4b1e941b5e97bf7d8218dc1c9b5ff088f1f813d8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
|
use std::str::FromStr;
pub fn get_num_threads() -> usize {
// Respond to the same environment variable as rayon.
match std::env::var("RAYON_NUM_THREADS")
.ok()
.and_then(|s| usize::from_str(&s).ok())
{
Some(x) if x > 0 => x,
Some(_) | None => num_cpus::get(),
}
}
|
