mod.rs 760 B

12345678910111213141516171819202122232425262728
  1. #![allow(dead_code)]
  2. #![allow(clippy::module_name_repetitions, clippy::shadow_unrelated)]
  3. use rayon::ThreadPoolBuilder;
  4. use std::env;
  5. pub mod eq;
  6. pub mod parse;
  7. /// Read the `ABORT_AFTER_FAILURE` environment variable, and parse it.
  8. pub fn abort_after() -> usize {
  9. match env::var("ABORT_AFTER_FAILURE") {
  10. Ok(s) => s.parse().expect("failed to parse ABORT_AFTER_FAILURE"),
  11. Err(_) => usize::max_value(),
  12. }
  13. }
  14. /// Configure Rayon threadpool.
  15. pub fn rayon_init() {
  16. let stack_size = match env::var("RUST_MIN_STACK") {
  17. Ok(s) => s.parse().expect("failed to parse RUST_MIN_STACK"),
  18. Err(_) => 20 * 1024 * 1024,
  19. };
  20. ThreadPoolBuilder::new()
  21. .stack_size(stack_size)
  22. .build_global()
  23. .unwrap();
  24. }