c - 如何编程让不同的进程在不同的 CPU 内核上运行？

Question

我正在编写一个带有 2 个进程的 linux C 程序。我将在不同的机器上运行该程序。

这些机器可能有多个 CPU 内核。

当我运行程序时，系统会为不同的进程分配不同的CPU内核吗？还是我需要编写一些代码以充分利用 CPU 内核？

Answer 1

如果您希望将线程/进程固定到特定的 CPU，那么您必须为此使用sched_setaffinity(2)系统调用或pthread_setaffinity_np(3)库调用。Linux 中的每个内核都有自己的虚拟 CPU ID。

这些调用允许您设置允许的 CPU 掩码。

否则，将取决于调度程序的题外话来运行你的线程，感觉就像在运行它们。

两者都不能保证您的进程并行运行。这是只有调度程序才能决定的事情，除非您实时运行。

这是一些示例代码：

#include <sched.h>

int run_on_cpu(int cpu) {

    cpu_set_t allcpus;

    CPU_ZERO(&allcpus);
    sched_getaffinity(0, sizeof(cpu_set_t), &allcpus);

    int num_cpus = CPU_COUNT(&allcpus);

    fprintf(stderr, "%d cpus available for scheduling\nAvailable CPUs: ", num_cpus);

    size_t i;
    for (i = 0; i < CPU_SETSIZE; i++) {
        if (CPU_ISSET(i, &allcpus))
            fprintf(stderr, "%zu, ", i);
    }
    fprintf(stderr, "\n");

    if (CPU_ISSET(cpu, &allcpus)) {
        cpu_set_t cpu_set;
        CPU_ZERO(&cpu_set);
        CPU_SET(cpu, &cpu_set);
        return pthread_setaffinity_np(pthread_self(), sizeof(cpu_set_t), &cpu_set);
    }

    return -1;
}