CPU 亲和性掩码（将线程放在不同的 CPU 上）

Python

什么是CPU亲和性掩码

CPU亲和性掩码是一种用于控制线程在不同CPU上运行的机制。当一个多线程的应用程序运行在多个CPU上时，有时候我们希望将某些线程固定在特定的CPU上执行，以提高性能或避免不必要的上下文切换。CPU亲和性掩码允许我们为线程分配特定的CPU，并通过设置亲和性掩码来控制线程在不同CPU上的调度。

如何使用CPU亲和性掩码

在使用CPU亲和性掩码之前，我们需要先了解一下系统中有多少个CPU可用。我们可以通过查询操作系统提供的API来获取CPU的数量。一旦得到了CPU的数量，我们可以使用一个二进制数来表示每个线程可以运行的CPU。

例如，如果我们的系统有4个CPU可用，我们可以用一个4位的二进制数来表示每个线程的亲和性掩码。每个位代表一个CPU，位的值为1表示该线程可以在该CPU上运行，位的值为0表示该线程不会在该CPU上运行。

下面是一个简单的示例代码，展示了如何使用CPU亲和性掩码将线程固定在不同的CPU上运行：

Python
import threading
import os
# 获取系统的CPU数量
num_cpus = os.cpu_count()
# 定义一个线程类
class MyThread(threading.Thread):
    def __init__(self, cpu_mask):
        super().__init__()
        self.cpu_mask = cpu_mask
    def run(self):
        # 设置线程的亲和性掩码
        os.sched_setaffinity(0, self.cpu_mask)
        # 执行线程的任务
        print(f"Thread {self.name} is running on CPU {os.sched_getaffinity(0)}")
# 创建4个线程，分别绑定到4个不同的CPU上运行
for i in range(num_cpus):
    # 将线程的亲和性掩码设置为对应的CPU
    cpu_mask = 1 << i</p>    thread = MyThread(cpu_mask)
    thread.start()
# 等待所有线程执行完毕
for thread in threading.enumerate():
    if thread != threading.current_thread():
        thread.join()

为线程分配不同的CPU

在上面的示例代码中，我们创建了4个线程，并将它们分别绑定到4个不同的CPU上运行。首先，我们获取系统的CPU数量，然后定义一个线程类MyThread，该类继承自threading.Thread。在线程类的run方法中，我们首先使用os.sched_setaffinity函数将线程的亲和性掩码设置为指定的CPU，然后打印出线程当前所在的CPU。

在主程序中，我们使用一个循环来创建并启动4个线程。对于每个线程，我们将其亲和性掩码设置为对应的CPU。最后，我们使用threading.enumerate()函数来等待所有线程执行完毕。

通过以上的代码，我们可以确保每个线程都会在指定的CPU上运行，从而实现了CPU亲和性掩码的控制。

CPU亲和性掩码是一种用于控制线程在不同CPU上运行的机制。通过设置线程的亲和性掩码，我们可以将线程固定在特定的CPU上执行，以提高性能或避免不必要的上下文切换。在实际应用中，我们可以根据系统的CPU数量和线程的需求，灵活地分配线程到不同的CPU上运行，以达到最佳的性能和资源利用率。