0

这是神经网络上反向传播算法代码的一部分。

在我们的例子中，我们想要并行化 for( pt=0; pt< N_PT_pair; pt++) 循环，for(epoch=0; epoch< MaxEpoch; epoch++) 不能并行化。

initialize W1[ ] [ ] and W2[ ][ ] with random values
for(epoch=0; epoch<MaxEpoch; epoch++)
     dW1[ ][ ]=0.0; dW2[ ][ ]=0.0; //sum of weight corrections
     sse = 0; // Sum of square of errors
     for( pt=0; pt<N_PT_pair; pt++)
          input = pattern[pt];
          compute output  // procedure as above
          compare target[pt] and output and
          compute dW2[ ][ ] += ... // procedure to be described
          compute dW1[ ][ ] += ... // procedure to be described
          for(k=1; k<=Noutput; k++) 
              sse+=pow((target[pt][k]-output[k]),2);
     end pt for loop   
     cout << "mean square error" << sse/N_PT_pair;
     W1[ ][ ] += rate*dW1[ ][ ]
     W2[ ][ ] += rate*dW2[ ][ ]
end epoch for loop

这些是分配和释放数组的代码

double** allocate_matrix(int rows,int cols) 
{    
    double **a;

    a = new double*[rows];    
    if(a==NULL){cout<<"matrix allocation failed"<<endl;exit(-1);}

    for (int j=0;j<rows;j++){   
       a[j] =  new double[cols];
       if(a[j]==NULL) {cout<<"matrix allocation failed"<<endl;exit(-1);}    
    }

    return a;
}

int deallocate_matrix(double**a,int rows)    
{
    for(int i=0;i<rows;i++)
       delete [] a[i];
    delete [ ] a;    
    return 0;
}

你能帮我们并行化代码吗？

4

1 回答 1

1

如果内部循环中的迭代相互独立，那么您可以简单地从一个 OpenMP 构造开始：

#pragma omp parallel for private(input,k) reduction(+:sse)
for( pt=0; pt<N_PT_pair; pt++)
     input = pattern[pt];
     compute output  // procedure as above
     compare target[pt] and output and
     compute dW2[ ][ ] += ... // procedure to be described
     compute dW1[ ][ ] += ... // procedure to be described
     for(k=1; k<=Noutput; k++) 
         sse+=pow((target[pt][k]-output[k]),2);
end pt for loop

dW1如果在一次以上的迭代中没有更新或更新任何元素，这将非常dW2有用，否则将需要原子访问，这会降低性能（OpenMP 仍然不支持在 C/C++ 中减少数组）。

如果您有大量的网络权重，您也可以以相同的方式并行化乘法。

并行区域可以进一步移到外循环之外，以减少 OpenMP 开销，single或者master可以使用 OpenMP 指令来隔离只应在单个线程中运行的代码。

为了让编译器理解#pragma omp指令，您必须启用 OpenMP 支持。具体如何完成取决于编译器：

  • -fopenmp海合会
  • -openmp用于英特尔 C/C++ 编译器
  • -xopenmp对于 Oracle Solaris Studio
  • MS Visual Studio 的项目属性 -> 等
于 2012-05-29T12:28:31.557 回答