Linux内核：进程管理：CPU绑定技术

1. 概念

CPU绑定指的是在多CPU的系统中将进程或线程绑定到指定的CPU核上去执行。在Linux中，我们可以利用CPU affinity属性把进程绑定到一个或多个CPU核上。

CPU Affinity是进程的一个属性，这个属性指明了进程调度器能够把这个进程调度到哪些CPU上。该属性要求进程在某个指定的 CPU 上尽量长时间地运行而不被迁移到其他处理器。

CPU Affinity分为2种：soft affinity和hard affinity。soft affinity只是一个建议，如果不可避免，调度器还是会把进程调度到其它的CPU上去执行；hard affinity则是调度器必须遵守的规则， 2.6 以上版本的 Linux 内核可以让开发人员可以编程实现hard affinity 。

2. 使用hard affinity的意义

提高CPU缓存命中率

CPU之间是不共享缓存的，如果进程频繁地在多个CPU之间切换，则会使旧CPU的cache失效，失去了利用 CPU 缓存的优势。如果进程只在某个CPU上执行，可以避免进程在一个CPU上停止执行，然后在不同的CPU上重新执行时发生的缓存无效而引起的性能成本。

适合对时间敏感的应用

在实时性要求高应用中，我们可以把重要的系统进程绑定到指定的CPU上，把应用进程绑定到其余的CPU上。这种做法确保对时间敏感的应用程序可以得到运行，同时可以允许其他应用程序使用其余的计算资源。

内核资料直通车：Linux内核源码技术学习路线+视频教程代码资料

学习直通车：Linux内核源码/内存调优/文件系统/进程管理/设备驱动/网络协议栈

3. 进程与CPU的绑定

在Linux中，用结构体cpu_set_t来表示CPU Affinity掩码，同时定义了一系列的宏来用于操作进程的可调度CPU集合：

#define _GNU_SOURCE

#include <sched.h>

void CPU_ZERO(cpu_set_t *set);

void CPU_SET(int cpu, cpu_set_t *set);

void CPU_CLR(int cpu, cpu_set_t *set);

int CPU_ISSET(int cpu, cpu_set_t *set);

int CPU_COUNT(cpu_set_t *set);

………………

具体的作用如下：

PU_ZERO()：清除集合的内容，让其不包含任何CPU。

CPU_SET()：添加cpu到集合中。

CPU_CLR()：从集合中移除cpu

CPU_ISSET() ：测试cpu是否在集合中。

CPU_COUNT()：返回集合中包含的CPU数量。

在Linux中，可以使用以下两个函数设置和获取进程的CPU Affinity属性：

#define _GNU_SOURCE

#include <sched.h>

int sched_setaffinity(pid_t pid, size_t cpusetsize,const cpu_set_t *mask);

int sched_getaffinity(pid_t pid, size_t cpusetsize,cpu_set_t *mask);

另外可以通过下面的函数获知当前进程运行在哪个CPU上：

int sched_getcpu(void);

如果调用成功，该函数返回一个非负的CPU编号值。

例程：

#define _GNU_SOURCE

#include <sched.h>

#include <unistd.h>

#include <sys/types.h>

#include <sys/wait.h>

#include <stdio.h>

#include <stdlib.h>

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

{

cpu_set_t set;

int parentCPU, childCPU;

int j;

int cpu_num = -1;

if (argc != 3) {

fprintf(stderr, "Usage: %s parent-cpu child-cpu\n", argv[0]);

exit(EXIT_FAILURE);

}

parentCPU = atoi(argv[1]);

childCPU = atoi(argv[2]);

CPU_ZERO(&set);

switch (fork()) {

case -1: { /* Error */

fprintf(stderr, "fork error\n");

exit(EXIT_FAILURE);

}

case 0: { /* Child */

CPU_SET(childCPU, &set);

if (sched_setaffinity(getpid(), sizeof(set), &set) == -1) {

fprintf(stderr, "child sched_setaffinity error\n");

exit(EXIT_FAILURE);

}

sleep(1);

if (-1 != (cpu_num = sched_getcpu())) {

fprintf(stdout, "The child process is running on cpu %d\n", cpu_num);

}

exit(EXIT_SUCCESS);

}

default: { /* Parent */

CPU_SET(parentCPU, &set);

if (sched_setaffinity(getpid(), sizeof(set), &set) == -1) {

fprintf(stderr, "parent sched_setaffinity error\n");

exit(EXIT_FAILURE);

}

if (-1 != (cpu_num = sched_getcpu())) {

fprintf(stdout, "The parent process is running on cpu %d\n", cpu_num);

}

wait(NULL); /* Wait for child to terminate */

exit(EXIT_SUCCESS);

}

}

}

程序首先用CPU_ZERO清空CPU集合，然后调用fork()函数创建一个子进程，并调用sched_setaffinity()函数给父进程和子进程分别设置CPU Affinity，输入参数parentCPU和 childCPU分别指定父进程和子进程运行的CPU号。指定父进程和子进程运行的CPU为1和0，程序输出如下：

# ./affinity_test 1 0
The parent process is running on cpu 1
The child process is running on cpu 0

4. 线程与CPU的绑定

前面介绍了进程与CPU的绑定，那么线程可不可以与CPU绑定呢？当然是可以的。在Linux中，可以使用以下两个函数设置和获取线程的CPU Affinity属性：

#define _GNU_SOURCE
#include <pthread.h>
int pthread_setaffinity_np(pthread_t thread, size_t cpusetsize, const cpu_set_t *cpuset);
int pthread_getaffinity_np(pthread_t thread, size_t cpusetsize, cpu_set_t *cpuset);

例程：

#define _GNU_SOURCE
#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
static void *thread_start(void *arg)
{
......
struct thread_info *tinfo = arg;
thread = tinfo->thread_id;
CPU_ZERO(&cpuset);
CPU_SET(tinfo->thread_num, &cpuset);
s = pthread_setaffinity_np(thread, sizeof(cpu_set_t), &cpuset);
if (s != 0) {
handle_error_en(s, "pthread_setaffinity_np");
}
CPU_ZERO(&cpuset);
s = pthread_getaffinity_np(thread, sizeof(cpu_set_t), &cpuset);
if (s != 0) {
handle_error_en(s, "pthread_getaffinity_np");
}
for (j = 0; j < cpu_num; j++) {
if (CPU_ISSET(j, &cpuset)) { //如果当前线程运行在CPU j上，则输出信息
printf(" thread %d is running on cpu %d\n", tinfo->thread_num, j);
}
}
pthread_exit(NULL);
}
int main(int argc, char *argv[])
{
......
cpu_num = sysconf(_SC_NPROCESSORS_CONF); //获取系统的CPU数量
tinfo = calloc(cpu_num, sizeof(struct thread_info));
if (tinfo == NULL) {
handle_error_en(0, "calloc");
}
for (j = 0; j < cpu_num; j++) { //有多少个CPU就创建多少个线程
tinfo[j].thread_num = j;
s = pthread_create(&tinfo[j].thread_id, NULL, thread_start, &tinfo[j]);
if (s != 0) {
handle_error_en(s, "pthread_create");
}
}
for (j = 0; j < cpu_num; j++) {
s = pthread_join(tinfo[j].thread_id, NULL);
if (s != 0) {
handle_error_en(s, "pthread_join");
}
}
......
}

程序首先获取当前系统的CPU数量cpu_num，然后根据CPU数量的数量创建线程，有多少个CPU就创建多少个线程，每个线程都运行在不同的CPU上。在虚拟机(4核)中运行结果如下：

$ ./thread_affinity

thread 1 is running on cpu 1

thread 0 is running on cpu 0

thread 3 is running on cpu 3

thread 2 is running on cpu 2

5. 用taskset命令实现CPU绑定

Linux 的taskset命令用于设置或检索由pid指定的运行进程的CPU Affinity，或者以给定的CPU Affinity属性启动新的进程。CPU Affinity属性用位掩码来表示，其中最低位对应第一逻辑CPU，最后一位与最后一个逻辑CPU对应。检索到的掩码仅反映与物理系统上的CPU相对应的位。如果给出无效的掩码（即当前系统上没有对应的有效的CPU掩码），则返回错误。掩码通常以十六进制形式给出。例如

0x00000001 表示CPU #0,

0x00000003 表示CPU #0 和 #1,

0x0000000f 表示CPU #0 ~ #3

taskset命令的选项如下：

-a, --all-tasks

设置或检索所有由pid指定的进程的CPU Affinity属性。

-c, --cpu-list numbers

指定处理器的数值列表，而不是位掩码。数字用逗号分隔，可以包括范围。比如：0,5,8-11。

-p, --pid

操作由pid指定的进程，不启动新的进程。

下面以Ubuntu16.04中的taskset命令说明该命令的使用方法：

显示进程运行的CPU
命令：taskset -p 1

结果：pid 1‘s current affinity mask: f

说明：f表示进程1运行在CPU#0~CPU#3上

指定进程运行在某个特定的CPU上
命令：taskset -cp 1,2 7737

结果：pid 7737's current affinity list: 0-3

pid 7737's new affinity list: 1,2

说明：该操作把进程7737限定在CPU#1~CPU#2上运行。

进程启动时指定CPU
命令：taskset -c 1-2 ./get_affinity

结果：This process is running on cpu 1

This process is running on cpu 2

说明：get_affinity程序通过sched_getaffinity()函数获取当前进程的CPU Affinity属性并输出提示信息。

6. 总结

本文通过几个例程详细介绍了Linux环境下进程或线程与CPU进行绑定的方法，具有较大的参考价值。如果觉得本文有用，欢迎点赞、收藏。

原文作者：精通Linux内核

原文地址：Linux内核：进程管理：CPU绑定技术 - 知乎（版权归原文作者所有，侵权留言联系删除）