在某些场合,我们想知道自己的进程到底执行在哪个cpu上或者想指定某一个进程在指定的cpu上执行(尤其是在多核场合),基本方法如下:
1、头文件
#define __USE_GNU
#include <unistd.h>
#include <sched.h>
#include <pthread.h>
#include <sys/syscall.h>
2、源码
static void __setAffinity_CPU_0(pid_t tid)
{
cpu_set_t cs;
CPU_ZERO(&cs);
CPU_SET(0,&cs);
sched_setaffinity(tid,sizeof(cs),&cs);
}
static void __setAffinity_CPU_1(pid_t tid)
{
cpu_set_t cs;
CPU_ZERO(&cs);
CPU_SET(1,&cs);
sched_setaffinity(tid,sizeof(cs),&cs);
}
static void __setAffinity_CPU_2(pid_t tid)
{
cpu_set_t cs;
CPU_ZERO(&cs);
CPU_SET(2,&cs);
sched_setaffinity(tid,sizeof(cs),&cs);
}
static void __setAffinity_CPU_3(pid_t tid)
{
cpu_set_t cs;
CPU_ZERO(&cs);
CPU_SET(3,&cs);
sched_setaffinity(tid,sizeof(cs),&cs);
}
3、使用方法
在调用子函数前执行添加以下代码可以指定子函数在指定的cpu上执行:
pid_t tid=syscall(SYS_gettid);
__setAffinity_CPU_0(tid);