如何使用 OpenMP 任务实现减少？

Question

我有这个执行简单归约的 OpenMP 代码：

for(k = 0; k < m; k++) 
{
      #pragma omp parallel for private(i) reduction(+:mysum) schedule(static) 
      for (i = 0; i < m; i++)
      {
           mysum += a[i][k] * a[i][k];
      }
}

我想创建一个与此代码等效的代码，但使用的是 OpenMP 任务。这是我通过关注this 文章所尝试的：

for(k = 0; k < m; k++) 
{
    #pragma omp parallel reduction(+:mysum)
    {
         #pragma omp single 
         {
                  for (i = 0; i < m; i++) 
                  {
                        #pragma omp task private(i) shared(k)
                        {
                                partialSum += a[i][k] * a[i][k];
                        }
                   }
         }

         #pragma omp taskwait
         mysum += partialSum;
     }
 }

变量partialSum被声明为threadprivate，它也是一个全局变量：

int partialSum = 0;
#pragma omp threadprivate(partialSum)

a 是一个简单的整数数组 (m x m)。

问题是当我多次运行上面的代码（带有任务的代码）时，我得到了不同的结果。

你知道我应该改变什么来完成这项工作吗？

非常感谢

Answer 1

private 变量未初始化（至少未由它们的外部值初始化）。 i 应该是 firstprivate。

如果你只是去掉private(i) shared(k)，默认情况下一切都是正确的。 k 来自parallel 部分之外，因此在parallel 部分中隐含shared。这也使得它在任务生成结构中隐含地shared。现在i 也是共享/共享的。如果您改为在本地定义它 (for (int i...)，它会隐式地变为 private 到 parallel 部分，因此在任务生成构造中隐式地变为 firstprivate。

你也应该添加

#pragma omp atomic
mysum += partialSum;

另一方面，您不一定需要taskwait（请参阅this answer）

请注意，谈话正确地使用了firstprivate。