mirror of https://github.com/vasi/pixz
Compare commits
2 Commits
f1b1b5f8af
...
d2131cc19f
Author | SHA1 | Date |
---|---|---|
Dave Vasilevsky | d2131cc19f | 2 years ago |
Wessel Dankers | 6a9443d955 | 2 years ago |
@ -1,5 +1,28 @@
|
|||||||
|
#define _GNU_SOURCE
|
||||||
|
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include "config.h"
|
||||||
|
|
||||||
|
#ifdef HAVE_SCHED_GETAFFINITY
|
||||||
|
|
||||||
|
#include <sched.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
|
size_t num_threads(void) {
|
||||||
|
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
|
||||||
|
return CPU_COUNT(&cpu_set);
|
||||||
|
}
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
size_t num_threads(void) {
|
size_t num_threads(void) {
|
||||||
return sysconf(_SC_NPROCESSORS_ONLN);
|
return sysconf(_SC_NPROCESSORS_ONLN);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
Loading…
Reference in New Issue