Retrieve the number of online CPUs on OpenBSD and NetBSD (#1916)

* Retrieve the number of online CPUs on OpenBSD and NetBSD * Spacing adjustment --------- Co-authored-by: dominic <[email protected]>
google · Feb 6, 2025 · 41e81b1 · 41e81b1
1 parent 47bc26c
commit 41e81b1
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/sysinfo.cc b/src/sysinfo.cc
@@ -153,11 +153,11 @@ ValueUnion GetSysctlImp(std::string const& name) {
   int mib[2];
 
   mib[0] = CTL_HW;
-  if ((name == "hw.ncpu") || (name == "hw.cpuspeed")) {
+  if ((name == "hw.ncpuonline") || (name == "hw.cpuspeed")) {
     ValueUnion buff(sizeof(int));
 
-    if (name == "hw.ncpu") {
-      mib[1] = HW_NCPU;
+    if (name == "hw.ncpuonline") {
+      mib[1] = HW_NCPUONLINE;
     } else {
       mib[1] = HW_CPUSPEED;
     }
@@ -482,7 +482,13 @@ std::string GetSystemName() {
 int GetNumCPUsImpl() {
 #ifdef BENCHMARK_HAS_SYSCTL
   int num_cpu = -1;
-  if (GetSysctl("hw.ncpu", &num_cpu)) return num_cpu;
+  constexpr auto* hwncpu =
+#ifdef HW_NCPUONLINE
+      "hw.ncpuonline";
+#else
+      "hw.ncpu";
+#endif
+  if (GetSysctl(hwncpu, &num_cpu)) return num_cpu;
   PrintErrorAndDie("Err: ", strerror(errno));
 #elif defined(BENCHMARK_OS_WINDOWS)
   SYSTEM_INFO sysinfo;