mirror of https://github.com/vasi/pixz
Compare commits
5 Commits
670ddc7c90
...
294875b108
Author | SHA1 | Date |
---|---|---|
Dave Vasilevsky | 294875b108 | 6 months ago |
Dave Vasilevsky | ebeb225a07 | 6 months ago |
Dave Vasilevsky | 7ed15497be | 6 months ago |
Dave Vasilevsky | 446087ed7a | 6 months ago |
Dave Vasilevsky | cb961a54b0 | 6 months ago |
@ -1,28 +1,35 @@
|
||||
#define _GNU_SOURCE
|
||||
|
||||
#include <unistd.h>
|
||||
|
||||
#include "config.h"
|
||||
|
||||
#ifdef HAVE_SCHED_GETAFFINITY
|
||||
|
||||
#include <sched.h>
|
||||
#include <stdio.h>
|
||||
#endif
|
||||
|
||||
#ifdef HAVE_SYSCONF
|
||||
#include <unistd.h>
|
||||
#endif
|
||||
|
||||
#ifdef HAVE_GETSYSTEMINFO
|
||||
#include <sysinfoapi.h>
|
||||
#endif
|
||||
|
||||
size_t num_threads(void) {
|
||||
#ifdef HAVE_SCHED_GETAFFINITY
|
||||
cpu_set_t cpu_set;
|
||||
CPU_ZERO(&cpu_set);
|
||||
|
||||
if (sched_getaffinity(0, sizeof cpu_set, &cpu_set) == -1)
|
||||
return sysconf(_SC_NPROCESSORS_ONLN);
|
||||
else
|
||||
if (sched_getaffinity(0, sizeof cpu_set, &cpu_set) == 0)
|
||||
return CPU_COUNT(&cpu_set);
|
||||
}
|
||||
|
||||
#else
|
||||
#endif
|
||||
|
||||
size_t num_threads(void) {
|
||||
#ifdef HAVE_SYSCONF
|
||||
return sysconf(_SC_NPROCESSORS_ONLN);
|
||||
}
|
||||
|
||||
#elif HAVE_GETSYSTEMINFO
|
||||
SYSTEM_INFO sysinfo;
|
||||
GetSystemInfo(&sysinfo);
|
||||
return sysinfo.dwNumberOfProcessors;
|
||||
#else
|
||||
#warning "No processor-detection enabled! Assuming 2 CPUs"
|
||||
return 2;
|
||||
#endif
|
||||
}
|
||||
|
Loading…
Reference in New Issue