From 99460cf759cb027075108bae193c4ebda9f400a3 Mon Sep 17 00:00:00 2001
From: Hiroki Noda <kubo39@gmail.com>
Date: Fri, 15 Jun 2018 03:04:30 +0900
Subject: [PATCH] Fallback if num processors via affinity mask is 0
---
src/lib.rs | 16 +++++++++-------
1 file changed, 9 insertions(+), 7 deletions(-)
diff --git a/src/lib.rs b/src/lib.rs
index 830e3b5..eb64218 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -339,15 +339,17 @@ fn get_num_cpus() -> usize {
count += 1
}
}
- count as usize
- } else {
- let cpus = unsafe { libc::sysconf(libc::_SC_NPROCESSORS_ONLN) };
- if cpus < 1 {
- 1
- } else {
- cpus as usize
+ if count > 0 {
+ return count as usize;
}
}
+
+ let cpus = unsafe { libc::sysconf(libc::_SC_NPROCESSORS_ONLN) };
+ if cpus < 1 {
+ 1
+ } else {
+ cpus as usize
+ }
}
#[cfg(any(