Linux下进程的CPU配置与线程绑定过程

#include <IOStream>

int main()
{
    while (true) {
        // 在这里编写你的代码
        // 例如，输出一条信息
        std::cout << "Hello, I am running in an infinite loop!" << std::endl;
    }

    return 0;
}

taskset -p < PID >

#include <unistd.h>

int sysconf(_SC_NPROCESSORS_CONF);/* 返回系统可以使用的核数，但是其值会包括系统中禁用的核的数目，因 此该值并不代表当前系统中可用的核数 */
int sysconf(_SC_NPROCESSORS_ONLN);/* 返回值真正的代表了系统当前可用的核数 */

/* 以下两个函数与上述类似 */
#include <sys/sysinfo.h>

int get_nprocs_conf (void);/* 可用核数 */
int get_nprocs (void);/* 真正的反映了当前可用核数 */

sched_setaffinity可以将某个进程绑定到一个特定的CPU。

#define _GNU_SOURCE             /* See feature_test_MACros(7) */
#include <sched.h>

/* 设置进程号为pid的进程运行在mask所设定的CPU上
 * 第二个参数cpusetsize是mask所指定的数的长度
 * 通常设定为sizeof(cpu_set_t)

 * 如果pid的值为0,则表示指定的是当前进程 
 */
int sched_setaffinity(pid_t pid, size_t cpusetsize, cpu_set_t *mask);

int sched_getaffinity(pid_t pid, size_t cpusetsize, cpu_set_t *mask);/* 获得pid所指示的进程的CPU位掩码,并将该掩码返回到mask所指向的结构中 */

/*

*该程序演示了如何使用sched_setaffinity函数将线程绑定到特定的CPU核心上运行。
*程序首先创建了两个线程，然后使用sched_setaffinity函数将线程1绑定到CPU 0上，将线程2绑定到CPU 3上。
*运行时，可以通过查看输出的pid来确定程序的进程ID。
*然后，程序将CPU_ZERO宏应用于一个cpu_set_t类型的变量mask，以将其初始化为空集。
*接下来，程序将CPU_SET宏应用于mask，将CPU 0和CPU 3添加到集合中。
*最后，程序调用sched_setaffinity函数将mask应用于当前进程，将线程1绑定到CPU 0上，将线程2绑定到CPU 3上。
*线程创建成功后，程序使用pthread_join函数等待线程1和线程2的结束。
*/
#define _GNU_SOURCE             /* See feature_test_macros(7) */
#include <stdio.h>
#include <sys/typOEDbePes.h>
#include <unistd.h>
#include <sched.h>
#include <pthread.h>
 
void* testfunc(void* t) {
  while(1);
  return NULL; 
}
 
int main()
{
  cpu_set_t mask; // 定义cpu_set_t类型的变量mask，用于存储CPU集合
  printf("pid=%d\n", getpid()); // 打印进程ID
  CPU_ZERO(&mask); // 将mask初始化为空集
  CPU_SET(0, &mask);//将cpu0绑定到mask中
  CPU_SET(3, &mask);//将cpu3绑定到mask中

  // 将mask应用于当前进程，绑定线程到指定的CPU核心
  sched_setaffinity(0, sizeof(cpu_set_t), &mask) ;
  
  pthread_t tid1;//创建线程1
  if (pthread_create(&tid1, NULL, (void *)testfunc, NULL) != 0) 
  {      
    fprintf(stderr, "thread create failed\n"); // 线程创建失败，打印错误信息
    return -1;   
  }
  pthread_t tid2;//创建线程2
  if (pthread_create(&tid2, NULL, (void *)testfunc, NULL) != 0) 
  {      
    fprintf(stderr, "thread create failed\n"); // 线程创建失败，打印错误信息
    return -1;   
  } 
  pthread_join(tid1, NULL); // 等待线程1结束
  pthread_join(tid1, NULL); // 等待线程2结束
  return 0;
}

CPU_ZERO() www.chinasem.cn清空一个集合
CPU_SET()与CPU_CLR()分别对将一个给定的CPU号加到一个集合或者从一个集合中去掉
CPU_ISSET()检查一个CPU号是否在这个集合中

sched_setaffinity(pid_t pid, unsigned int cpusetsize, cpu_set_t *mask)

sched_getaffinity(pid_t pid, unsigned int cpusetsize, cpu_set_t *mask)

// 此代码不完整，只是帮助理解绑定过程
// 在创建线程时添加以下代码，可以将该线程绑定到1核
cpu_set_t mask;

// 将掩码清零
CPU_ZERO(&mask);

// 将1添加到掩码中
CPU_SET(1, &mask);

// #将本线程绑定到1核
sched_setaffinity(0, sizeof(cpu_set_t), &mask);

/*
 *该程序演示了如何使用pthread_setaffinity_np函数将线程绑定到特定的CPU核心上运行。
 *程序首先创建了两个线程，然后使用pthread_setaffinity_np函数将线程1绑定到CPU 0上，将线程2绑定到CPU 3上。
 *运行时，可以通过查看输出的pid来确定程序的进程ID。
 *然后，程序将CPU_ZERO宏应用于一个cpu_set_t类型的变量mask，以将其初始化为空集。
 *接下来，程序使用pthread_create函数创建线程1和线程2，并检查线程创建是否成功。
 *然后，程序打印出线程1和线程2的ID。
 *程序使用CPU_SET宏将CPU 0添加到mask中，并使用pthread_setaffinity_np函数将mask应用于线程1，将线程1绑定到CPU 0上。
 *然后，程序清除之前设置的mask，并将CPU 3添加到mask中，并使用pthread_setaffinity_np函数将mask应用于线程2，将线程2绑定到CPU   3上。
 *最后，程序使用pthread_join函数等待线程1和线程2的结束。
*/
#define _GNU_SOURCE             /* See feature_test_macros(7) */
#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#include <sched.h>
#include <pthread.h>

void* testfunc(void* t) {
  int i = 3; // 初始化循环计数器为3
  while(i) { // 进入循环，条件为i非零
     sleep(5); // 休眠5秒
     printf("tid=%d,cpu=%d\n",pthread_self(), sched_getcpu()); // 打印线程ID和CPU编号
     i--; // 计数器减一
  }
  while(1); // 进入无限循环
  return NULL; 
}

int main()
{
  cpu_set_t mask; // 定义CPU集合
  printf("pid=%d\n", getpid()); // 打印进程ID
  CPU_ZERO(&mask); // 清空CPU集合
  
  pthread_t tid1; // 定义线程tid1
  if (pthread_create(&tid1, NULL, (void *)testfunc, NULL) != 0) 
  {      
    fprintf(stderr, "thread create failed\n"); // 线程创建失败，打印错误信息
    return -1;   
  }
  pthread_t tid2; // 定义线程tid2
  if (pthread_create(&tid2, NULL, (void *)testfunc, NULL) != 0) 
  {      
    fprintf(stderr, "thread create failed\n"); // 线程创建失败，打印错误信息
    return -1;   
  } 
  printf("tid1=%d,tid2=%d\n", tid1,tid2); // 打印线程tid1和tid2的值
 
  CPU_SET(0, &mask); // 将CPU0加入CPU集合
  pthread_setaffinity_np(tid1, sizeof(cpu_set_t), &mask) ; // 设置线程tid1的CPU亲和性为CPU0
  
  // 清除之前设置，重新设置绑定cpu3
  CPU_ZERO(&mask); // 清空CPU集合
  CPU_SET(3, &mask); // 将CPU3加入CPU集合
  pthread_setaffinity_np(tid2, sizeof(cpu_set_t), &mask) ; // 设置线程tid2的CPU亲和性为CPU3
  
  pthread_join(tid1, NULL); // 等待线程tid1结束
  pthread_join(tid1, NULL); // 等待线程tid2结束
  return 0;
}