使用动态调度计算openmp for循环的迭代次数

Question

我正在运行带有动态负载平衡的openmp for for循环。我想打印每个线程在程序结束时处理多少个任务/迭代。 循环如下所示：

chunk=1; 
#pragma omp parallel for schedule(dynamic,chunk) private(i) 
for(i=0;i<n;i++){ 
//loop code 
} 

Answer 1

没有什么更容易。刚刚拆分合并parallel for指令分成两个独立的结构，它允许您之前添加额外的代码，并在循环后：

#pragma omp parallel 
{ 
    int iters = 0; 
    #pragma omp for schedule(dynamic,chunk) 
    for (int i = 0; i < n; i++) { 
     ... 
     iters++; 
    } 
    #pragma omp critical 
    printf("Thread %d did %d iterations\n", omp_get_thread_num(), iters); 
} 

Answer 2

对每个线程使用一个私人计数器。没有其他办法。

喜欢的东西

int workload[n_threads] = {0, ...}; 
#pragma omp parallel for schedule(dynamic) private(i) 
for(int i=0;i<n;i++){ 
    //loop code 
    workload[omp_get_thread_num()]++; 
} 

Answer 3

如果你真的想在你的程序中的外底打印迭代次数您并行区域或其他代码（并避免虚假共享），简单的解决方案是使用threadprivate。

#include <stdio.h> 
#include <omp.h> 

int iters; 
#pragma omp threadprivate(iters) 

int main(void) { 
    omp_set_dynamic(0); //Explicitly turn off dynamic threads 
    int i; 
    int n = 10000; 
    #pragma omp parallel for schedule(dynamic) 
    for(i=0; i<n; i++) { 
    iters++; 
    } 
    #pragma omp parallel 
    #pragma omp critical 
    printf("Thread %d did %d iterations\n", omp_get_thread_num(), iters); 
} 

这是一个复杂的解决方案，它还要求您更改代码的结构。

#include <stdio.h> 
#include <stdlib.h> 
#include <omp.h> 

int main(void) { 
    int i; 
    int n = 100; 
    int nthreads; 
    int *aiters; 
    #pragma omp parallel 
    { 
    nthreads = omp_get_num_threads(); 
    #pragma omp single 
    aiters = malloc(sizeof *aiters * nthreads); 
    int iters = 0; 
    #pragma omp for schedule(dynamic) 
    for(i=0; i<n; i++) { 
     iters++; 
    } 
    aiters[omp_get_thread_num()]=iters; 
    } 
    for(i=0; i<nthreads; i++) 
    printf("Thread %d did %d iterations\n", i, aiters[i]); 
    free(aiters); 
}