您可以在 OpenMP 中将特定线程 ID 分组到唯一的任务组中吗？

Can you group specific thread id's into unique task groups in OpenMP?

本文关键字：唯一 任务 ID 线程 OpenMP      更新时间：2023-10-16

在运行我的程序之前，我这样做：

export KMP_AFFINITY=explicit,proclist=[0-47],granularity=fine

然后我生成了 48 个线程：

#pragma omp parallel num_threads(48) 
{
int id = omp_get_thread_num();
}

并行构造中的每个线程现在都有一个与处理器 ID 匹配的 ID。

现在。。。

处理器0 和处理器 24 实际上位于同一内核上。处理器 24 到 47 是 0 到 23 的超线程。

我有一个线程正在使用的工作对象向量。他们根据自己的 id 从向量中选择一个对象。

我正在尝试获取 0/24、1/25、2/26 等......超线程内核以配对特定的"worker"对象，然后在对象中使用任务组：

class Worker {
public:
int wId;
Worker(int i) {
wId = i;
}
void doWork() {
// Can I make a task group for the 2 threads reaching this code together?
// will barriers, taskgroups, critical pragmas wait for all 48 threads?
}
};
int main() {
vector<Worker> workers;
for (int i = 0; i < 48; ++i) {
Worker w(i);
workers.push_back(w);
}
#pragma omp parallel num_threads(48) 
{
int id = omp_get_thread_num();
workers[id % 24].doWork();
}
}

我不确定这样的事情是否有效。我的假设是，在该doWork()函数中使用taskgroup或barrier或critical编译指示都会影响所有 48 个线程，而不仅仅是应该共享对象的 2 个线程。

这是真的吗？如何创建一组已知处理器绑定的线程(即将线程 0 和线程 23 放入一个组中)以执行我想要的操作？

对任何感兴趣的人...正如@Gilles指出的那样，解决方案是嵌套。诀窍是正确导出：

export OMP_NESTED=1
export OMP_MAX_ACTIVE_LEVELS=2
export KMP_HOT_TEAMS=1
export KMP_HOT_TEAMS_MAX_LEVEL=2
export OMP_PROC_BIND=spread,spread
export OMP_PLACES=cores

然后你可以这样做：

#pragma omp num_thread(24) {
#pragma omp num_thread(2) {
// paired hyperthread code
}
}

(来源：https://software.intel.com/en-us/articles/process-and-thread-affinity-for-intel-xeon-phi-processors-x200)

相关文章：