Linux CPU线程绑核

为了加快程序的运行速度和充分利用CPU资源,我们可以人为将不同线程绑定在不同的cup上,例如有两个线程A,B,其中A已经在CPU0上运行,并且CPU0上还有其他的任务,那么我们可以将线程B绑到CPU1上,这样就可以减轻CPU0的负担,从而充分利用多核CPU。原来是一个CPU做两件事,现在两个CPU同时做两个事,使效率更高。

看下下面的代码

cpp 复制代码
#define _GNU_SOURCE
#include <sched.h>      //这两个头文件是一起的
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
 
void *myfun(void *arg)
{
    cpu_set_t mask;
    cpu_set_t get;
  
    CPU_ZERO(&mask);
    CPU_SET(1, &mask);
    if (pthread_setaffinity_np(pthread_self(), sizeof(mask), &mask) < 0) 
    {
        fprintf(stderr, "set thread affinity failed\n");
    }
 
    CPU_ZERO(&get);
 
    if (pthread_getaffinity_np(pthread_self(), sizeof(get), &get) < 0) 
    {
        fprintf(stderr, "get thread affinity failed\n");
    }
    if (CPU_ISSET(1, &get)) 
    {
            printf("thread %lu is running in processor 1\n", (long unsigned)pthread_self());
    }
    pthread_exit(NULL);     //退出线程
}
 
int main(int argc, char *argv[])
{
    pthread_t tid;          //用来创建新的线程
    cpu_set_t mask;         //用来设置
    cpu_set_t get;          //用来获
 
    int num = sysconf(_SC_NPROCESSORS_CONF);    //获取CPU核数
    printf("system has %d processor(s)\n", num);
 
    if (pthread_create(&tid, NULL, (void *)myfun, NULL) != 0)    //创建线程  myfun为入口函数
    {
        fprintf(stderr, "thread create failed\n");
        return -1;
    }
    
    CPU_ZERO(&mask);            //初始化某个CPU集,设置为空
    CPU_SET(0, &mask);          //将某个CPU加入到这个CPU集合里,这里是,也可以理解为,绑定CPU,这里是CPU0
    if (pthread_setaffinity_np(pthread_self(), sizeof(mask), &mask) < 0)       //pthread_self():获得线程自身ID
    {                                                                       //设置某一线程运行在某个CPU上
        fprintf(stderr, "set thread affinity failed\n");
    }
    
    CPU_ZERO(&get);
    
    if (pthread_getaffinity_np(pthread_self(), sizeof(get), &get) < 0) 
    {                                                                       //查看某一个CPU上有哪些线程
        fprintf(stderr, "get thread affinity failed\n");
    }
    if (CPU_ISSET(0, &get)) {                                               //判断某个CPU是不是在CPU集里,这里为CPU0
            printf("thread %lu is running in processor 0\n", (long unsigned)pthread_self());
    }
 
    pthread_join(tid, NULL);    //等待线程
    return 0;
}

编译一下

gcc cpu.c -pthread

thread 139946778752832 is running in processor 0

thread 139946770462464 is running in processor 1

可以看到我们将两个线程分别绑在了不同的CPU上面。

这里总结一下用到的基本函数:

cpp 复制代码
void CPU_ZERO (cpu_set_t *set);   //初始化,设为空

void CPU_SET (int cpu, cpu_set_t *set); //将某个cpu加入cpu集中 

void CPU_CLR (int cpu, cpu_set_t *set); //将某个cpu从cpu集中移出 

int CPU_ISSET (int cpu, const cpu_set_t *set); //判断某个cpu是否已在cpu集中设置了

int pthread_setaffinity_np(pthread_t thread,size_t cpusetsize,const cpu_set_t *cpuset); //设置CPU

int pthread_getaffinity_np(pthread_t thread,size_t cpusetsize, cpu_set_t *cpuset); //查看CPU
相关推荐
l1x1n05 小时前
Vim 编辑器常用操作详解(新手快速上手指南)
linux·编辑器·vim
ajassi20006 小时前
开源 python 应用 开发(三)python语法介绍
linux·python·开源·自动化
o不ok!6 小时前
Linux面试问题-软件测试
linux·运维·服务器
DaxiaLeeSuper7 小时前
Prometheus+Grafana+node_exporter监控linux服务器资源的方案
linux·grafana·prometheus
尽兴-8 小时前
如何将多个.sql文件合并成一个:Windows和Linux/Mac详细指南
linux·数据库·windows·sql·macos
kfepiza8 小时前
Netplan 中 bridges、bonds、ethernets、vlans 之间的关系 笔记250711
linux·tcp/ip·shell
小小不董8 小时前
深入理解oracle ADG和RAC
linux·服务器·数据库·oracle·dba
杰夫贾维斯9 小时前
CentOS Linux 8 的系统部署 Qwen2.5-7B -Instruct-AWQ
linux·运维·人工智能·机器学习·centos
kfepiza10 小时前
Netplan 配置网桥(Bridge)的模板笔记250711
linux·tcp/ip·ubuntu
kfepiza10 小时前
用Netplan配置网桥bridge笔记250711
linux·ubuntu·debian