Linux下多CPU中程序與CPU綁定的例子

現在多CPU的趨勢越來越大了. 有時候為了更好地操作機器, 需要将某個程序綁定到具體的CPU上去. 下面給出了一個程序綁定到具體的CPU上去的一個例子.  

#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

                } 

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

                        if (CPU_ISSET(i, &get)) 

                        { 

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

                        } 

        return 0; 

} 

下面是在兩個終端分别執行了./cpu 0  ./cpu 2 後得到的結果. 效果比較明顯.  

Cpu0  :  5.3%us,  5.3%sy,  0.0%ni, 87.4%id,  0.0%wa,  0.0%hi,  2.0%si,  0.0%st 

Cpu1  :  0.0%us,  0.0%sy,  0.0%ni,100.0%id,  0.0%wa,  0.0%hi,  0.0%si,  0.0%st 

Cpu2  :  5.0%us, 12.2%sy,  0.0%ni, 82.8%id,  0.0%wa,  0.0%hi,  0.0%si,  0.0%st 

Cpu3  :  0.0%us,  0.0%sy,  0.0%ni,100.0%id,  0.0%wa,  0.0%hi,  0.0%si,  0.0%st 

Cpu4  :  0.0%us,  0.0%sy,  0.0%ni, 99.7%id,  0.3%wa,  0.0%hi,  0.0%si,  0.0%st 

Cpu5  :  0.0%us,  0.0%sy,  0.0%ni,100.0%id,  0.0%wa,  0.0%hi,  0.0%si,  0.0%st 

Cpu6  :  0.0%us,  0.0%sy,  0.0%ni,100.0%id,  0.0%wa,  0.0%hi,  0.0%si,  0.0%st 

Cpu7  :  0.0%us,  0.0%sy,  0.0%ni,100.0%id,  0.0%wa,  0.0%hi,  0.0%si,  0.0%st