如何将进程、线程与CPU核进行绑定
概念
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。
使用hard affinity的意义
- 提高CPU缓存命中率
CPU各核之间是不共享缓存的，如果进程频繁地在多个CPU核之间切换，则会使旧CPU核的cache失效，失去了利用CPU缓存的优势。如果进程只在某个CPU上执行，可以避免进程在一个CPU上停止执行，然后在不同的CPU上重新执行时发生的缓存无效而引起的性能成本。
- 适合对时间敏感的应用
在实时性要求高应用中，我们可以把重要的系统进程绑定到指定的CPU上，把应用进程绑定到其余的CPU上。这种做法确保对时间敏感的应用程序可以得到运行，同时可以允许其他应用程序使用其余的计算资源。
如何将进程与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);具体的作用如下：
CPU_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如何将线程与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用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 ~ #3taskset命令的选项如下：
-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属性并输出提示信息。总结
本文通过几个简单的例子介绍了Linux环境下进程、线程与CPU的绑定方法，希望对大家有参考意义。
