c - 有什么干净的方法可以让 OpenMP pragmas 与宏一起工作？

Question

我必须将 OpenMP 添加到客户的代码中。它有几个像这样工作的宏:

int i, imax;
#ifdef MAC1
double x1, y1
#endif
#ifdef MAC2
double x2, y2
#endif

//first loop:
for (i=0; i<imax; i++ ) {
#ifdef MAC1
//process x1, y1
#endif
#ifdef MAC2
//process x2, y2
#endif
//do a ton of other stuff
}

// Lots more code.  No way it will all work in one omp region.

//second loop:
for (i=0; i<imax; i++ ) {
#ifdef MAC1
//process x1, y1
#endif
#ifdef MAC2
//process x2, y2
#endif
//do a ton of other stuff
}

完全有可能 MAC1 和 MAC2 都被定义，或者其中之一，或者两者都不定义。

现在，我想在 OpenMP 下运行循环。这通常是行不通的:

#pragma omp parallel private(x1, y1,   \
                             x2, y2,   \
                             and a     \
                             ton of    \
                             other stuff)
{
...
}

... 因为如果 MAC1 和/或 MAC2 未定义，编译器会提示 x1, y1 和/或 x2, y2 未定义。

我已经想到了几种解决这个问题的方法，但这些方法要么行不通，要么不合标准:

1) 我不能只将这些变量的定义拉入 omp 并行区域，因为正如我所说，我稍后必须在第二个区域中使用它们，我不能在那里重新定义它们。

2) 我想我可以将这些变量拉出它们的宏 block 并定义它们是否需要。这将是最简单的解决方案，但这似乎有点“hack-ish”。我怀疑原则上在某些情况下这是行不通的。

3) 我想我可以写几个 omp pragmas，每个宏组合一个，包含在四个不同的 #ifdefs 中，但是我需要四个不同的可选 pragmas，代码将得到凌乱真的很快。

4) 我试着做类似的事情

#pragma omp parallel private(the other stuff) \
#ifdef MAC1
                     private(x1, y1)   \
#endif
#ifdef MAC2
                     private(x2, y2)   \
#endif
{ ... }

这仍然不是很好看，但它比四个不同的可选 pragma 好得多。尽管如此，我还是尝试了几种语法变体，但编译器不知道我在尝试做什么。

有人知道一种干净的方法来做我在这里想做的事情吗？

Answer 1

我能想到的最简单、最自然的事情是:

// Just once, in a header if need be:

#ifdef MAC1
// Note trailing comma:
#define MAC1_VARS x1, y1,
#else
#define MAC1_VARS
#endif

#ifdef MAC2
// Note trailing comma:
#define MAC2_VARS x2, y2,
#else
#define MAC2_VARS
#endif
/////////////////////////

// ...

// each parallel region / loop:
#pragma omp parallel private(MAC1_VARS \
                         MAC2_VARS     \
                         other, stuff)
{
    // ...
}

MAC1_VARS 和 MAC2_VARS 因此会适本地展开，要么展开为空，要么展开为变量列表的适当子序列。

如果您可以修改有条件地声明这些变量的客户端代码，那么我会将 MAC1_VARS 和 MAC2_VARS 的定义也放在那里，而不是放在单独的预处理器条件中.