Linux C CPU绑核

AcTarjan

已于 2023-11-03 15:38:30 修改

阅读量320

点赞数 2

分类专栏： C语言文章标签： linux

于 2023-11-02 14:38:01 首次发布

本文链接：https://blog.csdn.net/AcTarjan/article/details/134181851

版权

C语言专栏收录该内容

18 篇文章 1 订阅

订阅专栏

查看CPU信息：lscpu 或 cat /proc/cpuinfo

CPU隔核

CPU隔核：指定的CPU核心只参与最低限度的OS内核计算。也就是OS尽量不会使用这些CPU核。
应用场景：在一些需要特定计算能力的场合，比如科学计算、3D图形渲染等需要大量计算的任务中，可以通过CPU隔核来优化计算效率。
如何配置：修改/etc/default/grub，在GRUB_CMDLINE_LINUX中添加如下内容。

#对cpu core1,2,7,8进行隔核
GRUB_CMDLINE_LINUX="isolcpus=1-2,7-8"	

#centos更新配置
grub2-mkconfig -o /boot/grub2/grub.cfg 

#ubuntu更新配置
update-grub

reboot重启

进程绑核

#define _GNU_SOURCE
#include <sched.h>

void set_affinity(int cord_id)
{
    long total_cpus = sysconf(_SC_NPROCESSORS_ONLN);
    printf("total cpus: %ld\n", total_cpus);
    //core index: 0 ~ total_cpus-1

    pid_t pid = getpid();

    cpu_set_t cpu_mask;
    CPU_ZERO(&cpu_mask);    // 初始化set集，将set置为空
    CPU_SET(cord_id, &cpu_mask);  //选择core_id

    int ret = sched_setaffinity(pid, sizeof(cpu_mask), &cpu_mask);
    if (ret != 0)
    {
        printf("fail to set CPU affinity, error:%s\n", strerror(errno));
    }

    cpu_set_t cpu_get;
    sched_getaffinity(pid, sizeof(cpu_get), &cpu_get);
    for (int i = 0; i < total_cpus; i++)
    {
        if ( CPU_ISSET(i, &cpu_get) )
            printf("%d set core %d successfully\n", pid, i);
    }
}

线程绑核

#define _GNU_SOURCE            
#include <pthread.h>

void set_pthread_affinity(int core_id)
{
	pthread_t tid = pthread_self();
	
	cpu_set_t cpu_mask;
    CPU_ZERO(&cpu_mask);    // 初始化set集，将set置为空
    CPU_SET(core_id, &cpu_mask);  //选择core_id
    int ret = pthread_setaffinity_np(tid, sizeof(cpu_set_t), &cpu_mask);
    if (ret != 0)
    {
        printf("fail to bind core\n");
    }

    cpu_set_t cpu_get;
    pthread_getaffinity_np(tid, sizeof(cpu_set_t), &cpu_get);
    if(CPU_ISSET(core_id, &cpu_get))
    {
       printf("pthread %lu set core %d successfully\n", tid, core_id);
    }
}

//this is a pthread
void *worker(void *arg)			//arg is core_id
{
	int core_id = *((int*)arg);
	set_pthread_affinity(core_id);
}