在Linux内核中使用“ sched_setaffinity（）”

Question

sched_setaffinity上有很多帖子，但是在内核空间中使用此帖子几乎没有。

我正在使用内核4.14.79。

我尝试使用用户空间方法以以下形式调用sched_setaffinity：

cpu_set_t my_set;        
CPU_ZERO(&my_set);       
CPU_SET(7, &my_set);     
sched_setaffinity(0, sizeof(cpu_set_t), &my_set);

但是在尝试编译内核时，我遇到了错误，并且意识到这种形式在内核空间中不起作用。

我看到sched_affinity是在sched.h中定义的，其格式为：

extern long sched_setaffinity(pid_t pid, const struct cpumask *new_mask);

但是我对如何使用正确的CPU编号正确创建new_mask参数感到困惑。关于它的文档不是很有帮助。

有人可以举例说明如何在内核空间中使用此功能来将进程设置为特定的CPU吗？

Answer 1

在浏览内核文件以寻找cpumask出现的位置后，自己找到了答案。

您可以使用以下两个功能：

cpumask_clear(struct cpumask *dstp) //clear the mask you are about to use

cpumask_set_cpu(unsigned int cpu, struct cpumask *dstp) //set the cpu number

所以我用的是这样的：

struct cpumask mask;  
cpumask_clear(&mask); 
cpumask_set_cpu(cpuNumber, &mask); 
sched_setaffinity(pid, &mask);