Linux下进程绑定多CPU运行

在服务器上，我们经常会有多个CPU的情况，而此时如果把进程都绑定在一个CPU上，那么对资源太多浪费了，下面的代码就实现了如何将程序绑定在不同的cpu上。传入参数代表绑定第几个cpu（从0开始计算）

//cpu_test.cpp

#include<stdlib.h>

#include<stdio.h>

#include<sys/types.h>

#include<sys/sysinfo.h>

#include<unistd.h>

//#define __USE_GNU

#include<sched.h>

#include<ctype.h>

#include<string.h>

int main(int argc, char* argv[])

{

int num = sysconf(_SC_NPROCESSORS_CONF);

int created_thread = 0;

int myid;

int i;

int j = 0;

cpu_set_t mask;

cpu_set_t get;

if (argc != 2)

{

printf(“usage : ./cpu num\n”);

exit(1);

}

myid = atoi(argv[1]);

printf(“system has %i processor(s). \n”, num);

CPU_ZERO(&mask);

CPU_SET(myid, &mask);

if (sched_setaffinity(0, sizeof(mask), &mask) == -1)

{

printf(“warning: could not set CPU affinity, continuing…\n”);

}

while (1)

{

usleep(10000);

CPU_ZERO(&get);

if (sched_getaffinity(0, sizeof(get), &get) == -1)

{

printf(“warning: cound not get cpu affinity, continuing…\n”);

}

for (i = 0; i < num; i++)

{

if (CPU_ISSET(i, &get))

{

printf(“this process %d is running processor : %d\n”,getpid(), i);

}

}

}

return 0;

}

//g++ cpu_test.cpp -o cpu_test

出处：http://blogread.cn/it/article/837?f=sa