162 lines
5.7 KiB
C
162 lines
5.7 KiB
C
/**
|
|
* \file affinity.h
|
|
* License details are found in the file LICENSE.
|
|
* \brief
|
|
* Macros used to set and get CPU affinity
|
|
* \author Taku Shimosawa <shimosawa@is.s.u-tokyo.ac.jp> \par
|
|
* Copyright (C) 2011 - 2012 Taku Shimosawa
|
|
*/
|
|
/*
|
|
* HISTORY:
|
|
*/
|
|
|
|
/*
|
|
* Adapted from Linux sched.h
|
|
*
|
|
* Modified to omit __GNUC_PREREQ checks and use
|
|
* the naive implementations.
|
|
*
|
|
*/
|
|
|
|
#if !defined __cpu_set_t_defined
|
|
# define __cpu_set_t_defined
|
|
/* Size definition for CPU sets. */
|
|
# define __CPU_SETSIZE 1024
|
|
# define __NCPUBITS (8 * sizeof (__cpu_mask))
|
|
|
|
/* Type for array elements in 'cpu_set_t'. */
|
|
typedef unsigned long int __cpu_mask;
|
|
|
|
/* Basic access functions. */
|
|
# define __CPUELT(cpu) ((cpu) / __NCPUBITS)
|
|
# define __CPUMASK(cpu) ((__cpu_mask) 1 << ((cpu) % __NCPUBITS))
|
|
|
|
/* Data structure to describe CPU mask. */
|
|
typedef struct
|
|
{
|
|
__cpu_mask __bits[__CPU_SETSIZE / __NCPUBITS];
|
|
} cpu_set_t;
|
|
|
|
/* Access functions for CPU masks. */
|
|
# define __CPU_ZERO_S(setsize, cpusetp) \
|
|
do { \
|
|
size_t __i; \
|
|
size_t __imax = (setsize) / sizeof (__cpu_mask); \
|
|
__cpu_mask *__bits = (cpusetp)->__bits; \
|
|
for (__i = 0; __i < __imax; ++__i) \
|
|
__bits[__i] = 0; \
|
|
} while (0)
|
|
# define __CPU_SET_S(cpu, setsize, cpusetp) \
|
|
(__extension__ \
|
|
({ size_t __cpu = (cpu); \
|
|
__cpu < 8 * (setsize) \
|
|
? (((__cpu_mask *) ((cpusetp)->__bits))[__CPUELT (__cpu)] \
|
|
|= __CPUMASK (__cpu)) \
|
|
: 0; }))
|
|
# define __CPU_CLR_S(cpu, setsize, cpusetp) \
|
|
(__extension__ \
|
|
({ size_t __cpu = (cpu); \
|
|
__cpu < 8 * (setsize) \
|
|
? (((__cpu_mask *) ((cpusetp)->__bits))[__CPUELT (__cpu)] \
|
|
&= ~__CPUMASK (__cpu)) \
|
|
: 0; }))
|
|
# define __CPU_ISSET_S(cpu, setsize, cpusetp) \
|
|
(__extension__ \
|
|
({ size_t __cpu = (cpu); \
|
|
__cpu < 8 * (setsize) \
|
|
? ((((__const __cpu_mask *) ((cpusetp)->__bits))[__CPUELT (__cpu)] \
|
|
& __CPUMASK (__cpu))) != 0 \
|
|
: 0; }))
|
|
|
|
# define __CPU_COUNT_S(setsize, cpusetp) \
|
|
__sched_cpucount (setsize, cpusetp)
|
|
|
|
# define __CPU_EQUAL_S(setsize, cpusetp1, cpusetp2) \
|
|
(__extension__ \
|
|
({ __const __cpu_mask *__arr1 = (cpusetp1)->__bits; \
|
|
__const __cpu_mask *__arr2 = (cpusetp2)->__bits; \
|
|
size_t __imax = (setsize) / sizeof (__cpu_mask); \
|
|
size_t __i; \
|
|
for (__i = 0; __i < __imax; ++__i) \
|
|
if (__bits[__i] != __bits[__i]) \
|
|
break; \
|
|
__i == __imax; }))
|
|
|
|
# define __CPU_OP_S(setsize, destset, srcset1, srcset2, op) \
|
|
(__extension__ \
|
|
({ cpu_set_t *__dest = (destset); \
|
|
__const __cpu_mask *__arr1 = (srcset1)->__bits; \
|
|
__const __cpu_mask *__arr2 = (srcset2)->__bits; \
|
|
size_t __imax = (setsize) / sizeof (__cpu_mask); \
|
|
size_t __i; \
|
|
for (__i = 0; __i < __imax; ++__i) \
|
|
((__cpu_mask *) __dest->__bits)[__i] = __arr1[__i] op __arr2[__i]; \
|
|
__dest; }))
|
|
|
|
# define __CPU_ALLOC_SIZE(count) \
|
|
((((count) + __NCPUBITS - 1) / __NCPUBITS) * sizeof (__cpu_mask))
|
|
# define __CPU_ALLOC(count) __sched_cpualloc (count)
|
|
# define __CPU_FREE(cpuset) __sched_cpufree (cpuset)
|
|
|
|
#if 0
|
|
__BEGIN_DECLS
|
|
|
|
extern int __sched_cpucount (size_t __setsize, const cpu_set_t *__setp)
|
|
__THROW;
|
|
extern cpu_set_t *__sched_cpualloc (size_t __count) __THROW __wur;
|
|
extern void __sched_cpufree (cpu_set_t *__set) __THROW;
|
|
|
|
__END_DECLS
|
|
#endif
|
|
|
|
#endif
|
|
|
|
/* Access macros for `cpu_set'. */
|
|
# define CPU_SETSIZE __CPU_SETSIZE
|
|
# define CPU_SET(cpu, cpusetp) __CPU_SET_S (cpu, sizeof (cpu_set_t), cpusetp)
|
|
# define CPU_CLR(cpu, cpusetp) __CPU_CLR_S (cpu, sizeof (cpu_set_t), cpusetp)
|
|
# define CPU_ISSET(cpu, cpusetp) __CPU_ISSET_S (cpu, sizeof (cpu_set_t), \
|
|
cpusetp)
|
|
# define CPU_ZERO(cpusetp) __CPU_ZERO_S (sizeof (cpu_set_t), cpusetp)
|
|
# define CPU_COUNT(cpusetp) __CPU_COUNT_S (sizeof (cpu_set_t), cpusetp)
|
|
|
|
# define CPU_SET_S(cpu, setsize, cpusetp) __CPU_SET_S (cpu, setsize, cpusetp)
|
|
# define CPU_CLR_S(cpu, setsize, cpusetp) __CPU_CLR_S (cpu, setsize, cpusetp)
|
|
# define CPU_ISSET_S(cpu, setsize, cpusetp) __CPU_ISSET_S (cpu, setsize, \
|
|
cpusetp)
|
|
# define CPU_ZERO_S(setsize, cpusetp) __CPU_ZERO_S (setsize, cpusetp)
|
|
# define CPU_COUNT_S(setsize, cpusetp) __CPU_COUNT_S (setsize, cpusetp)
|
|
|
|
# define CPU_EQUAL(cpusetp1, cpusetp2) \
|
|
__CPU_EQUAL_S (sizeof (cpu_set_t), cpusetp1, cpusetp2)
|
|
# define CPU_EQUAL_S(setsize, cpusetp1, cpusetp2) \
|
|
__CPU_EQUAL_S (setsize, cpusetp1, cpusetp2)
|
|
|
|
# define CPU_AND(destset, srcset1, srcset2) \
|
|
__CPU_OP_S (sizeof (cpu_set_t), destset, srcset1, srcset2, &)
|
|
# define CPU_OR(destset, srcset1, srcset2) \
|
|
__CPU_OP_S (sizeof (cpu_set_t), destset, srcset1, srcset2, |)
|
|
# define CPU_XOR(destset, srcset1, srcset2) \
|
|
__CPU_OP_S (sizeof (cpu_set_t), destset, srcset1, srcset2, ^)
|
|
# define CPU_AND_S(setsize, destset, srcset1, srcset2) \
|
|
__CPU_OP_S (setsize, destset, srcset1, srcset2, &)
|
|
# define CPU_OR_S(setsize, destset, srcset1, srcset2) \
|
|
__CPU_OP_S (setsize, destset, srcset1, srcset2, |)
|
|
# define CPU_XOR_S(setsize, destset, srcset1, srcset2) \
|
|
__CPU_OP_S (setsize, destset, srcset1, srcset2, ^)
|
|
|
|
# define CPU_ALLOC_SIZE(count) __CPU_ALLOC_SIZE (count)
|
|
# define CPU_ALLOC(count) __CPU_ALLOC (count)
|
|
# define CPU_FREE(cpuset) __CPU_FREE (cpuset)
|
|
|
|
#if 0
|
|
/* Set the CPU affinity for a task */
|
|
extern int sched_setaffinity (__pid_t __pid, size_t __cpusetsize,
|
|
__const cpu_set_t *__cpuset) __THROW;
|
|
|
|
/* Get the CPU affinity for a task */
|
|
extern int sched_getaffinity (__pid_t __pid, size_t __cpusetsize,
|
|
cpu_set_t *__cpuset) __THROW;
|
|
#endif
|
|
|