进/线程绑核运行

进/线程绑核运行

我们在做性能摸底的时候需要排除多核处理器间的差异，甚至是频率的影响。有时特定业务为了确定性，甚至会单独划出一个CPU单独供给，那么我们可以怎么让某个进/线程绑定到特定的核心上运行呢？

taskset

如果不侵入式修改代码，我们可以使用 taskset 命令。它可以设置/获取已经运行的进程的亲和性，或者在给定亲和性的情况下拉起一个新进程。亲和性默认使用比特掩码来表示的，0x1表示 CPU #0，也可以使用CPU列表的形式，逗号分隔即可，如 0,5,8-11，它表示CPU #0 #5 #8 #9 #10 #11，同时还支持间隔指定，如 --cpu-list 0-10:2，它表示CPU #0 #2 #4 #6 #8 #10。

set or retrieve a process’s CPU affinity

# 默认行为是指定亲和性的情况下运行一个新进程
taskset mask command [arguments]

# 获取指定pid的亲和性
taskset -p pid

# 设置亲和性
taskset -p mask pid

# 设置的时候如果使用 cpu-list 的形式，则需要带上 -c 选项
taskset -pc cpu-list pid

# --cpu-list 选项只用于指定亲和性的情况下运行一个新进程
taskset --cpu-list cpu-list command

举例

$ taskset -p 1
pid 1's current affinity mask: ffff
$ taskset -pc 1
pid 1's current affinity list: 0-15

$ taskset -pc 1 2726
pid 2726's current affinity list: 0,1
pid 2726's new affinity list: 1

修改源码

如果可以侵入式修改源码，那么我们可以使用 sched_setaffinity 系统调用。sched_setaffinity可以将某个进程绑定到一个特定的CPU。
如果是希望绑定线程到某个CPU上运行，可以使用 pthread_setaffinity_np。