c++ - openMP 条件编译指示“if else”

Question

我有一个 for 循环，可以使用schedule(static)或schedule(dynamic, 10)根据条件执行。目前，我的代码不够 DRY（不要重复自己），并且为了适应以前的功能，它具有以下重复：

boolean isDynamic; //can be true or false
if(isDynamic){
    #pragma omp parallel for num_threads(thread_count) default(shared) private(...) schedule(dynamic, 10)
    for(...){
        //for code inside
    }
}else{
    #pragma omp parallel for num_threads(thread_count) default(shared) private(...) schedule(static)
    for(...){
        //SAME for code inside, in fact, this is the EXACT same for as before
    }
}

阅读这些线程后，我注意到 openMP 有一个#if(expression)编译指示：

但是，尽管我见过很多人遇到我的问题，但似乎缺乏一个通用的解决方案。最好的解决方案是将for循环的主体转换为一个函数，然后调用该函数，但是这种解决方案对我来说还不够好。

所以我想知道，OpenMP 是否有#if(expression) else某种编译指示？就像是：

#if(isDynamic )pragma omp parallel for num_threads(thread_count) default(shared) 
private(...) schedule(dynamic, 10) 
else 
pragma omp parallel for num_threads(thread_count) default(shared) 
private(...) schedule(static)

或者我是否被迫将我的 for 循环体放入一个单独的函数中并以这种方式调用它？

score 4 · Accepted Answer

这是个有趣的问题。基本上，您想schedule在运行时更改策略。据我所知，当前的 OpenMP 没有这样的指令。

我遇到了与您完全相同的问题。正如您所提到的，我的解决方案最终将循环体作为一个函数。否则，您需要使用丑陋的宏。

但是，我也尝试使用schedule(runtime)读取环境变量的OMP_SCHEDULE. 所以，我在运行时更改了这个环境变量，但没有奏效。这是因为 OpenMP 运行时在开始时只读取该环境一次。这可能是特定于实现的问题。因此，其他实现可能会即时读取此环境变量。你可以试试这个方法。

score 1 · Accepted Answer

这已经晚了几年，但对于这种特殊情况，您可以使用运行时库在运行时设置计划。它在 OpenMP 4.5 的 §3.2.12 中定义：

void omp_set_schedule(omp_sched_t kind, int chunk_size);
typedef enum omp_sched_t {
    omp_sched_static = 1,
    omp_sched_dynamic = 2,
    omp_sched_guided = 3,
    omp_sched_auto = 4
} omp_sched_t;

对于您的情况，您要做的是：

/* wherever currently isDynamic is set */
if (isDynamic) {
    omp_set_schedule(omp_sched_dynamic, 10);
} else {
     /* chunk_size < 1 uses default */
    omp_set_schedule(static, 0);
}

/* later */
#pragma omp parallel for num_threads(thread_count) default(shared) private(...) schedule(runtime)
for (...) {
   /* do a thing */
}

c++ - openMP 条件编译指示“if else”

2 回答 2

Related

Reference