如何配置OMP_PROC_BIND为"close"时使用的物理CoreID列表？

问题背景 #

我们正尝试在新硬件平台上使用OpenMP优化HPC应用，这类应用需要精确的核心绑定/固定，否则性能会减半。当前我们为每个平台提供自定义GOMP_CPU_AFFINITY映射，但这十分繁琐——不同硬件版本的映射不同，甚至不同固件版本的平台也会改变CoreID物理映射，用户无法实时检测这些变化。

我们希望HPC应用只需将GOMP_PROC_BIND设为"close"，OpenMP就能适配对应平台，但这需要硬件厂商定义各机器上"close"的具体含义。我们有意实现此功能，但不清楚OpenMP从何处/如何获取用于close、spread等模式的CoreID列表（受外部需求限制，该机器的CoreID空间模式对软件开发者而言完全随机）。

烦请告知OpenMP定义OMP_PROC_BIND所用CoreID列表的位置与方式，以便我们进行配置；若有需要，我们可为此平台定制修改源码后的OpenMP版本。

核心绑定逻辑的源码位置与修改方式（针对GCC GOMP实现） #

1. CPU拓扑信息的获取路径 #

GOMP依赖系统或平台特定接口获取CPU拓扑，以此为基础生成绑定用的CoreID列表：

• 通用Linux平台：读取/sys/devices/system/cpu/下的拓扑文件（如cpuX/topology/core_id、numa_node等）
• 自定义硬件平台：若系统默认接口无法识别你的拓扑布局，需修改拓扑探测代码

对应源码文件（GCC源码树内）：

• libgomp/config/linux/proc.c：Linux系统通用拓扑探测逻辑
• libgomp/config/{你的平台}/proc.c：若为特定平台，可能存在专属拓扑实现文件

2. OMP_PROC_BIND模式的CoreID列表生成逻辑 #

close/spread等模式的CoreID序列由拓扑排序逻辑决定：

• 默认close模式：优先将线程绑定在同一超线程组、物理核心、NUMA节点内
• 你需要修改排序规则，让GOMP按照硬件定义的“紧密”关系生成CoreID列表

关键源码文件：

• libgomp/team.c：线程组绑定核心逻辑，gomp_bind_threads函数负责核心绑定流程
• libgomp/cpu-set.c：CPU集合管理与排序，gomp_cpu_set_build等函数生成符合绑定模式的CoreID序列

3. 自定义平台适配步骤 #

针对CoreID映射随机的硬件，需完成以下修改：

• 替换拓扑探测逻辑：在平台专属proc.c中，添加自定义拓扑探测（如读取硬件寄存器、固件接口获取真实物理映射关系）
• 修改绑定模式排序规则：在cpu-set.c或team.c中，重写close模式的CoreID排序逻辑，按照硬件定义的“紧密”规则排列CoreID
• 编译定制版libgomp：将修改后的源码编译为动态库，让HPC应用链接此定制运行时

4. 验证绑定效果 #

修改完成后，可通过以下方式验证：

• 设置GOMP_DEBUG=1运行应用，查看输出的线程绑定详情
• 使用taskset或ps -eLo pid,tid,psr命令检查线程实际绑定的CoreID

内容的提问来源于stack exchange，提问作者user2465201