OpenMP 任意缩减(合并）

OpenMP arbitrary reduction (merge)

本文关键字：合并 任意缩 OpenMP      更新时间：2023-10-16

我想使用 OpenMP 收集一些数据。为此，我检查了许多候选人，只收集那些满足某些条件的候选人。简化的示例如下。

#include <bits/stdc++.h>
#include "omp.h"
using namespace std;
class DataPoint {};
DataPoint random_data_point() {
// generate random data point
}
bool test(DataPoint r) {
// do something
}
int main() {
constexpr int num_iterations = 10000;
set<DataPoint> good_points;
#pragma omp parallel for reduction(???)
for (int iter = 0; iter < num_iterations; iter++) {
DataPoint r = random_data_point();
if (test(DataPoint))
good_points.insert(r);
}
//  ...
return 0;
}

问题是如何有效地使用 OpenMP。我的想法是每个线程收集自己的数据(在上面的例子中 - 它自己的集合(，然后在所有线程完成后，它们的集合被合并。

注意 1：我的迭代非常独立(如上例所示(，因此 OpenMP 应该会有所帮助。 注2：在实际程序中，我使用的数据结构比set复杂。(实际上，它是数据点的凸包。

现代 OpenMP 允许您定义自己的归约函数，因此，如果您愿意，您应该能够以这种方式实现这一点，但实际上，对于这么简单的事情，沿着这些行编写代码可能会更容易(未经测试，键入此答案，未编译：-((

#pragma omp parallel
{
set<Data_Point> thread_good_points;
#pragma omp for nowait
for (int iter = 0; iter < num_iterations; iter++) {
DataPoint r = random_data_point();
if (test(DataPoint))
thread_good_points.insert(r);
}
#pragma omp critical
good_points.merge(thread_good_points);
}
相关文章：