linux cpu affinity demo

from: http://blog.csdn.net/chenji001/article/details/5546396

参考： http://www.ibm.com/developerworks/cn/linux/l-affinity.html http://www.ibm.com/developerworks/linux/library/l-affinity/index.html

代码如下：

#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)   
        {   
  
                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;   
} 

运行结果（需要在linux下编译运行，在unix-center ubuntu server上）：

/home/a/j/nomad2:./a.out 2 |head 
system has 2 processor(s). 
warning: could not set CPU affinity, continuing...
this process 20117 is running processor : 0
this process 20117 is running processor : 1
this process 20117 is running processor : 0
this process 20117 is running processor : 1
this process 20117 is running processor : 0
this process 20117 is running processor : 1
this process 20117 is running processor : 0
this process 20117 is running processor : 1