当前位置：首页 > news >正文

linux 进程/线程设置核亲和性

news 2025/7/18 8:20:21

1，线程绑定内核

#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>

// 定义一个函数，用于设置线程的CPU亲和性
void set_thread_affinity(pthread_t thread, int cpu_id) {
cpu_set_t cpuset;
int s;

// 清空CPU集合并添加指定的CPU
CPU_ZERO(&cpuset);
CPU_SET(cpu_id, &cpuset);

// 设置线程的CPU亲和性
s = pthread_setaffinity_np(thread, sizeof(cpu_set_t), &cpuset);
if (s != 0) {
perror("pthread_setaffinity_np");
exit(EXIT_FAILURE);
}

// 验证设置是否成功
s = pthread_getaffinity_np(thread, sizeof(cpu_set_t), &cpuset);
if (s != 0) {
perror("pthread_getaffinity_np");
exit(EXIT_FAILURE);
}

// 打印CPU亲和性设置结果
for (int i = 0; i < CPU_SETSIZE; i++) {
if (CPU_ISSET(i, &cpuset)) {
printf("Thread %lu is running on CPU %d\n", thread, i);
}
}
}

// 线程函数
void* thread_function(void* arg) {
// 设置本线程的CPU亲和性到CPU 0
set_thread_affinity(pthread_self(), 0);

// 执行线程的工作...
return NULL;
}

int main() {
pthread_t thread;

// 创建线程
if (pthread_create(&thread, NULL, thread_function, NULL) != 0) {
perror("pthread_create");
return 1;
}

// 等待线程结束
if (pthread_join(thread, NULL) != 0) {
perror("pthread_join");
return 1;
}

return 0;
}

设置本线程的CPU亲和性到CPU 0使用的函数是 set_thread_affinity(pthread_self(), 0)，使用CPU_ZERO和CPU_SET宏来初始化一个CPU集合并设置亲和性掩码。

2，进程绑定内核，绑定某个进程到某一个核（核亲和性）

#include <stdio.h>
#include <stdlib.h>
#include <sched.h>
#include <unistd.h>

int main() {
cpu_set_t cpuset;
pid_t pid;

// 获取当前进程的PID
pid = getpid();

// 清空CPU集合并设置亲和性掩码
CPU_ZERO(&cpuset);
// 假设我们要绑定到第0个CPU核心上
CPU_SET(0, &cpuset);

// 设置当前进程的CPU亲和性
if (sched_setaffinity(pid, sizeof(cpuset), &cpuset) == -1) {
perror("sched_setaffinity");
exit(EXIT_FAILURE);
}

// 打印进程ID和设置的CPU亲和性
printf("Process %d bound to CPU %d\n", pid, 0);

// 为了演示效果，让进程睡眠一段时间
sleep(60);

return EXIT_SUCCESS;
}