为OpenMP函数创建别名||部分禁用openMP

Question

是否可以为openMP名称创建预处理别名？

即

#define #myParallel #pragma omp parallel for
#define #myBarrier  #pragma omp barrier 

为了…目的

#ifdef MY_LIB_DISABLE_OPENMP
#define #myParallel 
#define #myBarrier
#else
#define #myParallel #pragma omp parallel for
#define #myBarrier  #pragma omp barrier 
#endif

我有两个库，都使用openmp。第一个库(并行化)调用第二个库(也是并行化的)中的函数，我相信这会导致指数级的线程数量。我希望能够禁用第二个库的openmp调用。(这两个库都是我的，所以我可以修补一下)。

唯一的另一种解决方案是复制/粘贴包装在#ifdef #else中的大量代码，但这似乎非常低效/非常不优雅。

Answer 1

OpenMP运行时不太可能创建指数级数量的线程。

一种更简单的方法是构建自己的库的两个版本

• libfoo.so (在没有 OpenMP)
• libfoo_omp.so的情况下构建(使用OpenMP构建OpenMP)

并链接到最适合的库。

注C在使用_OPENMP标志调用时定义了OpenMP宏(例如，带有GCC的-fopenmp )，如果其他一切都失败了，您可以使用它。

Answer 2

您有多种选择

1. 使用了现代OpenMP的基于任务的结构(任务循环等)。它们将自然地利用嵌套并行性，而无需创建新线程，因此不存在oversubscription.
2. Don't set OMP_NESTED的危险，在这种情况下，明显嵌套的内部并行性将被扁平化(这很可能是默认行为)。
3. 如果您必须使用C宏技巧，请阅读_Pragma，它允许您在宏中嵌入编译指示。