自定义OpenMP归约中omp_orig的问题分析与疑问

问题场景 #

使用Clang 15.0.0（arm64-apple-darwin23.5.0）搭配2024年7月通过brew安装的libomp时，遇到自定义OpenMP归约的异常行为：

• 初始化器中使用omp_priv = omp_orig时，串行运行输出正确结果(1000,1000)，但设置OMP_NUM_THREADS=4并行运行时输出错误值。
• 将初始化改为omp_priv = 0后，并行运行结果恢复正确，但对omp_orig的实际含义及该方案的潜在风险存在疑问。

关于omp_orig的行为解析 #

根据OpenMP规范，omp_orig定义为指向待归约原始变量的存储，但在Clang+libomp的具体实现中，它并非始终代表原始变量的初始值：

• 并行归约启动时，线程私有副本（omp_priv）的初始化阶段，omp_orig可能指向的是归约过程中已被修改的原始变量，或是内部临时中间值，而非程序启动时的初始基准值。
• 这种情况下，每个线程的私有副本初始化值不一致，最终导致归约结果混乱。

解决方案的合理性与注意事项 #

改用omp_priv = 0初始化是符合自定义归约逻辑的正确操作：

• 自定义归约的核心逻辑是：每个线程从归约操作的单位元开始累积计算，最终将所有线程的结果合并到原始变量。对于求和类归约，单位元就是0，这与OpenMP内置reduction(+:var)的行为完全一致。
• 该方案无明显潜在风险，前提是你的归约操作的单位元确实为0。如果是其他类型归约（如求积的单位元为1），需对应修改初始化值为对应单位元。

额外注意：

• 自定义归约的合并操作必须满足结合律，否则并行结果可能与串行结果不一致（这是OpenMP归约的通用要求，与初始化方式无关）。
• 若原始变量存在非零初始值，OpenMP会在归约结束后将合并后的线程结果与原始初始值合并，自定义归约需遵循此逻辑——即线程私有副本从单位元开始初始化，最终再与原始变量初始值合并。

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