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;
}

如何编程以便不同的进程在不同的CPU内核上运行？

1 个答案: