OpenMP C++ - Как распараллелить эту функцию?

Question

Я хотел бы распараллелить эту функцию, но я новичок с открытыми мками, и я был бы признателен, если кто-то может помочь мне:

void my_function(float** A,int nbNeurons,int nbOutput, float* p, float* amp){ 
    float t=0; 
    for(int r=0;r<nbNeurons;r++){ 
     t+=p[r]; 
    } 

    for(int i=0;i<nbOutput;i++){ 
     float coef=0; 
     for(int r=0;r<nbNeurons;r++){ 
     coef+=p[r]*A[r][i]; 
     } 
    amp[i]=coef/t; 
    } 
} 

Я не знаю, как распараллелить это правильно, так как из двойная петля для, на данный момент, я только подумал о том, чтобы сделать: #pragma omp parallel for reduction(+:t)

Но я думаю, что это не лучший способ быстрее получить вычисления через openMp.

Спасибо заранее,

Answer 1

Прежде всего: мы должны знать контекст. Где ваш профилировщик говорит вам, сколько времени потрачено?

В целом, крупнозернистая параллелистика работает лучше всего, так как @Alex сказал: parallellize внешний для петли.

void my_function(float** A,int nbNeurons,int nbOutput, float* p, float* amp) 
{ 
    float t=0; 
    for(int r=0;r<nbNeurons;r++) 
     t+=p[r]; 

#pragma parallel omp for 
    for(int i=0;i<nbOutput;i++){ 
     float coef=0; 
     for(int r=0;r<nbNeurons;r++){ 
      coef+=p[r]*A[r][i]; 
     } 
     amp[i]=coef/t; 
    } 
} 

В зависимости от фактических объемов, это может быть интересно вычислить т в фоновом режиме, и переместить разделение из параллельного цикла:

void my_function(float** A,int nbNeurons,int nbOutput, float* p, float* amp) 
{ 
    float t=0; 
#pragma omp parallel shared(amp) 
    { 
#pragma omp single nowait // only a single thread executes this 
     { 
      for(int r=0;r<nbNeurons;r++) 
       t+=p[r]; 
     } 

#pragma omp for 
     for(int i=0;i<nbOutput;i++){ 
      float coef=0; 
      for(int r=0;r<nbNeurons;r++){ 
       coef+=p[r]*A[r][i]; 
      } 
      amp[i]=coef; 
     } 

#pragma omp barrier 
#pragma omp master // only a single thread executes this 
     { 
      for(int i=0; i<nbOutput; i++){ 
       amp[i] /= t; 
      } 
     } 
    } 
} 

Примечания непроверенного кода. Иногда OMP имеет сложную семантику, поэтому я, возможно, пропустил там «общую» декларацию. Тем не менее, профайлер не будет уведомлять вас об этом быстро.