c++ - 待定:parallel_for 中的局部和全局结果

Question

我对并行循环中的局部值和更新全局变量有疑问。

例如，在伪代码中:我在一个很长的 vector 中搜索最大值。我可以像这样循环执行:

int max;
for(i ...) {
    if (max < vector[i]) max = vector[i];
}

我可以轻松地将它与 OpenMP 并行化:

int max;
#pragma omp parallel
{
    int local_max;
#pragma omp parallel for
    for(i ...) {
        if (local_max < vector[i]) local_max = vector[i];
    }

#pragma omp critical
    {   
        // choose the best solution from all
        if (max < local_max) max = local_max; local_max
    }
}

我如何在 TBB parallel_for 中做同样的事情？我不要求确切的代码，我只想知道如何在循环结束时而不是在每次迭代时更新全局结果...

(我是 TBB 新手)

Answer 1

您在本例中所做的称为缩减，因此请使用 parallel_reduce。它应该比在锁下更新全局变量更有效。基本思想是local_max是parallel_reduce的body类的成员变量，其join()方法接收body的另一个实例并将 local_max 更新为当前值和另一个实例中的值中的较大者。然后在调用 parallel_reduce 之后，您从原始主体对象中取出 local_max 值并将其分配给全局变量。