参考文章:Linux中线程与CPU核的绑定_bluenet13的博客-CSDN博客
long sysconf(int name);
函数sysconf获取运行时的系统配置信息
设置线程亲和性,将线程绑定到指定CPU核
int pthread_setaffinity_np(pthread_t thread, size_t cpusetsize,const cpu_set_t *cpuset);
thread:线程id
cpusetsize:集合内存大小
cpuset:CPU核的集合
获取指定线程的CPU集合
int pthread_getaffinity_np(pthread_t thread, size_t cpusetsize, cpu_set_t *cpuset);
thread:线程id
cpusetsize:集合内存大小
cpuset:CPU核的集合
对CPU核的集合操作
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集中设置了
#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#include <sched.h>
void *myfun(void *arg)
{
cpu_set_t mask;
cpu_set_t get;
char buf[256];
int i;
int j;
int num = sysconf(_SC_NPROCESSORS_CONF); // 在运行时获取配置信息,获取CPU核个数
CPU_ZERO(&mask); // 初始化,设为空
printf("system has %d processor(s)\n", num);
for(i = 0; i < num; i++){
CPU_SET(i, &mask); // 将某个cpu加入cpu集中
}
// 将线程绑定到cpu集中的各个CPU核,这样做不太好,会导致线程在各CPU核中切换,导致性能下降
// 本次只为测试
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");
}
for (i = 0; i < num; i++) {
if (CPU_ISSET(i, &get)) {
printf("thread %ld is running in processor %d\n", pthread_self(), i);
}
j = 0;
while (j++ < 100000000){
memset(buf, 0, sizeof(buf));
}
}
pthread_exit(NULL);
}
int main(){
int num = sysconf(_SC_NPROCESSORS_CONF);
printf("CPU核数:%d \n", num);
pthread_t tid;
if (pthread_create(&tid, NULL, myfun, NULL) != 0) {
fprintf(stderr, "thread create failed\n");
return -1;
}
pthread_join(tid, NULL);
return 0;
}
在终端执行操作:
[root@chen02 CPUBind]# g++ -o cpu CPU_Bind.cpp -lpthread