Linux 中用c++实现线程绑定CPU

2022-12-16 815

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： Linux 中用c++实现线程绑定CPU

前言

嵌入式里面我们会使用到多核的cpu，随着产品芯片性能提升，我们也会有很多功能，以及很多进程产生运行，这个时候我们在任务调度调优的时候，把一些进程绑定到固定cpu运行，下面就来分享一下cpu绑定运行的过程：

首先运行的环境需要多核，大家可以按照下面命令进行查询对应设备的cpu数量

查看cpu有几个核

使用cat /proc/cpuinfo查看cpu信息，如下两个信息：

processor，指明第几个cpu处理器

cpu cores，指明每个处理器的核心数

基本概念

cpu亲和性(affinity)

CPU的亲和性，就是进程要在指定的 CPU 上尽量长时间地运行而不被迁移到其他处理器，也称为CPU关联性；再简单的点的描述就将指定的进程或线程绑定到相应的cpu上；在多核运行的机器上，每个CPU本身自己会有缓存，缓存着进程使用的信息，而进程可能会被OS调度到其他CPU上，如此，CPU cache命中率就低了，当绑定CPU后，程序就会一直在指定的cpu跑，不会由操作系统调度到其他CPU上，性能有一定的提高。

软亲和性(affinity)

就是进程要在指定的 CPU 上尽量长时间地运行而不被迁移到其他处理器，Linux 内核进程调度器天生就具有被称为软 CPU 亲和性(affinity) 的特性，这意味着进程通常不会在处理器之间频繁迁移。这种状态正是我们希望的，因为进程迁移的频率小就意味着产生的负载小。

硬亲和性(affinity)

简单来说就是利用linux内核提供给用户的API，强行将进程或者线程绑定到某一个指定的cpu核运行。

代码绑定实现

上面我们使用cat /proc/cpuinfo命令查询了自己设备的CPU，以我为例，我的电脑从0～7一共有8核。

下面是代码的demo：

#include <stdio.h>
#include <unistd.h>
#include <thread>
void thread_func1()
{
        cpu_set_t mask;
        CPU_ZERO(&mask);
        CPU_SET(1, &mask); //指定该线程使用的CPU
        if (pthread_setaffinity_np(pthread_self(), sizeof(mask), &mask) < 0) {
                perror("pthread_setaffinity_np");
        }
        int count = 0;
        while(1)
        {
                count ++;
                sleep(1);
                printf("fun 1 cnt :%d \n",count);
                for(int i = 0; i < 8; i++) {
                        if (CPU_ISSET(i, &mask))  //查看cpu i 是否在get 集合当中
                        {
                                printf("1 this process %d of running processor: %d\n", getpid(), i);
                        }
                }
        }
}
void thread_func2()
{
        int count = 0;
        cpu_set_t mask;
        CPU_ZERO(&mask);
        CPU_SET(5, &mask);
        if (pthread_setaffinity_np(pthread_self(), sizeof(mask), &mask) < 0) {
                perror("pthread_setaffinity_np");
        }
        while(1)
        {
                usleep(1000*1000);
                count ++;
                printf("fun 2 cnt :%d \n",count);
                for(int i = 0; i < 8; i++) {
                 if (CPU_ISSET(i, &mask))  //查看cpu i 是否在get 集合当中
                {
                       printf("2 this process %d of running processor: %d\n", getpid(), i);
                    }
                }
        }
}
int main(int argc, char *argv[])
{
      int cpus = 0;
        cpus = sysconf(_SC_NPROCESSORS_CONF);
        printf("cpus: %d\n", cpus); //查看cpu的个数；
    cpu_set_t mask;
    CPU_ZERO(&mask);
    CPU_SET(7, &mask);
    if (sched_setaffinity(0, sizeof(mask), &mask) < 0) {
        perror("sched_setaffinity");
    }
    std::thread t1(thread_func1);
    std::thread t2(thread_func2);
        usleep(1000); /* 让当前的设置有足够时间生效*/
       while(1)
        {
                /*查看运行在当前进程的cpu*/
                sleep(1); /* 让当前的设置有足够时间生效*/
                printf("fun main \n");
                for(int i = 0; i < cpus; i++) {
                        if (CPU_ISSET(i, &mask))  //查看cpu i 是否在get 集合当中
                        {
                                printf("3 this process %d of running processor: %d\n", getpid(), i);
                        }
                }
        }
    t1.join();
    t2.join();
}

使用此命令编译：

g++ test_select_cpu.cpp -pthread --std=c++11

上面一共运行了三个线程，一个是main 主线程，还有两个是自己定义的thread。

最重要的设置代码就是下面所示：设置cpu 亲和

cpu_set_t mask;
CPU_ZERO(&mask);/* 初始化set集，将set置为空*/
CPU_SET(5, &mask);/* 将对应的cpu序号加入到集合*/
if (pthread_setaffinity_np(pthread_self(), sizeof(mask), &mask) < 0) /*设置cpu 亲和性(affinity)*/
{ 
    perror("pthread_setaffinity_np");
}