CPU 亲合力就是指在 Linux 系统中能够将一个或多个进程绑定到一个或多个处理器上运行.
一个进程的 CPU 亲合力掩码决定了该进程将在哪个或哪几个 CPU 上运行。在一个多处理器系统中,设置 CPU 亲合力的掩码可能会获得更好的性能.
一个 CPU 的亲合力掩码用一个 cpu_set_t 结构体来表示一个 CPU 集合,下面的几个宏分别对这个掩码集进行操作:
- CPU_ZERO() 清空一个集合
- CPU_SET() 与 CPU_CLR() 分别对将一个给定的 CPU 号加到一个集合或者从一个集合中去掉.
- CPU_ISSET() 检查一个 CPU 号是否在这个集合中.
下面两个函数就是用来设置获取线程 CPU 亲和力状态:
sched_setaffinity(pid_t pid, unsigned int cpusetsize, cpu_set_t *mask)
该函数设置进程为 pid 的这个进程,让它运行在 mask 所设定的 CPU 上。如果 pid 的值为 0,则表示指定的是当前进程,使当前进程运行在 mask 所设定的那些 CPU 上。第二个参数 cpusetsize 是 mask 所指定的数的长度,通常设定为 sizeof(cpu_set_t)。如果当前 pid 所指定的进程此时没有运行在 mask 所指定的任意一个 CPU 上,则该指定的进程会从其它 CPU 上迁移到 mask 的指定的一个 CPU 上运行。
sched_getaffinity(pid_t pid, unsigned int cpusetsize, cpu_set_t *mask)
该函数获得 pid 所指示的进程的 CPU 位掩码,并将该掩码返回到 mask 所指向的结构中,即获得指定 pid 当前可以运行在哪些 CPU 上。同样,如果 pid 的值为 0,也表示的是当前进程。
#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>
#include <pthread.h>
#define THREAD_MAX_NUM 200 // 1 个 CPU 内的最多进程数
int num = 0; //cpu中核数
void *threadFun(void *arg) // arg 传递线程标号(自己定义)
{
cpu_set_t mask; // CPU 核的集合
cpu_set_t get; // 获取在集合中的 CPU
int *a = (int *)arg;
int i;
printf("the thread is:%d\n", *a); // 显示是第几个线程
CPU_ZERO(&mask); // 置空
CPU_SET(*a, &mask); // 设置亲和力值
if (sched_setaffinity(0, sizeof(mask), &mask) == -1) // 设置线程 CPU 亲和力
{
printf("warning: could not set CPU affinity, continuing...\n");
}
CPU_ZERO(&get);
if (sched_getaffinity(0, sizeof(get), &get) == -1) // 获取线程 CPU 亲和力
{
printf("warning: cound not get thread affinity, continuing...\n");
}
for (i = 0; i < num; i++) {
if (CPU_ISSET(i, &get)) // 判断线程与哪个 CPU 有亲和力
{
printf("this thread %d is running processor : %d\n", i, i);
}
}
return NULL;
}
int main(int argc, char *argv[])
{
int tid[THREAD_MAX_NUM];
int i;
pthread_t thread[THREAD_MAX_NUM];
num = sysconf(_SC_NPROCESSORS_CONF); // 获取核数
if (num > THREAD_MAX_NUM) {
printf("num of cores[%d] is bigger than THREAD_MAX_NUM[%d]!\n", num, THREAD_MAX_NUM);
return -1;
}
printf("system has %i processor(s). \n", num);
for (i = 0; i < num; i++) {
tid[i] = i; // 每个线程必须有个 tid[i]
pthread_create(&thread[i], NULL, threadFun, (void *)&tid[i]);
}
for (i = 0; i < num; i++) {
pthread_join(thread[i], NULL); // 等待所有的线程结束,线程为死循环所以 CTRL+C 结束
}
return 0;
}
(SAW:Game Over!)