Linux / 线程绑定到 CPU 指定核心

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;
}

转载：c++ 线程绑定到cpu指定核心 - Wangtn - 博客园

（SAW：Game Over！）