X-Git-Url: https://git.lttng.org/?p=urcu.git;a=blobdiff_plain;f=tests%2Fapi_gcc.h;h=17487d5c701e81b66349b798b7f7fade5b1d8a96;hp=872c873f9760da2b812c54aa19a2dc4d0c7894f0;hb=d8540fc53386cd7ff270085aca621e32772832ba;hpb=f540a0fdac1fb71bba34c6aafff55f0e2fce7877 diff --git a/tests/api_gcc.h b/tests/api_gcc.h index 872c873..17487d5 100644 --- a/tests/api_gcc.h +++ b/tests/api_gcc.h @@ -2,6 +2,8 @@ #ifndef _INCLUDE_API_H #define _INCLUDE_API_H +#include "../config.h" + /* * common.h: Common Linux kernel-isms. * @@ -461,13 +463,25 @@ static void wait_all_threads(void) } } +#ifndef HAVE_CPU_SET_T +typedef unsigned long cpu_set_t; +# define CPU_ZERO(cpuset) do { *(cpuset) = 0; } while(0) +# define CPU_SET(cpu, cpuset) do { *(cpuset) |= (1UL << (cpu)); } while(0) +#endif + static void run_on(int cpu) { +#if HAVE_SCHED_SETAFFINITY cpu_set_t mask; CPU_ZERO(&mask); CPU_SET(cpu, &mask); +#if SCHED_SETAFFINITY_ARGS == 2 + sched_setaffinity(0, &mask); +#else sched_setaffinity(0, sizeof(mask), &mask); +#endif +#endif /* HAVE_SCHED_SETAFFINITY */ } /*