cpu-affinity-wrapper.c

   1 #define _GNU_SOURCE
   2 #include <sched.h>
   3
   4 int cpu_setsize = CPU_SETSIZE;
   5 int cpu_mask_size = sizeof(cpu_set_t);
   6
   7 int get_cpu_affinity_mask(cpu_set_t *mask)
   8 {
   9     return sched_getaffinity(0, CPU_SETSIZE, mask);
  10 }
  11
  12 int set_cpu_affinity_mask(cpu_set_t *mask)
  13 {
  14     return sched_setaffinity(0, CPU_SETSIZE, mask);
  15 }
  16
  17 void zero_cpu_affinity_mask(cpu_set_t *mask)
  18 {
  19     CPU_ZERO(mask);
  20 }
  21
  22 void set_cpu_affinity(int cpu, cpu_set_t *mask)
  23 {
  24     CPU_SET(cpu, mask);
  25 }
  26
  27 int cpu_affinity_p(int cpu, cpu_set_t *mask)
  28 {
  29     return CPU_ISSET(cpu, mask);
  30 }
  31
  32 void clear_cpu_affinity(int cpu, cpu_set_t *mask)
  33 {
  34     CPU_CLR(cpu, mask);
  35 }
  36
  37
  38