Вычислить число итераций в цикле openmp с динамическим расписанием

Question

Я запускаю цикл for с openmp с динамической балансировкой нагрузки. Я хотел бы напечатать, сколько задач/итераций обрабатывается каждый поток в конце программы. Цикл выглядит следующим образом:

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); 
}