提交: 748479b4f5705859d117caa2d7c03ee0a26d07c6 - MolDS (git)

Commit MetaInfo

• 修订版: 748479b4f5705859d117caa2d7c03ee0a26d07c6 (tree)
• 时间: 2013-08-01 10:51:52
• 作者: Mikiya Fujii <mikiya.fujii@gmai...>
• Commiter: Mikiya Fujii

Log Message

Calculation of force on ground state for MNDO-series is MPI parallelized. #31814

git-svn-id: https://svn.sourceforge.jp/svnroot/molds/trunk@1451 1136aad2-a195-0410-b898-f5ea1d11b9d8

更改概述

• modified: src/mndo/Mndo.cpp
• modified: src/zindo/ZindoS.cpp

差异

--- a/src/mndo/Mndo.cpp
+++ b/src/mndo/Mndo.cpp
@@ -2267,23 +2267,29 @@ void Mndo::CalcForceExcitedTwoElecPart(double* force,
 // electronicStateIndex is index of the electroinc eigen state.
 // "electronicStateIndex = 0" means electronic ground state. 
 void Mndo::CalcForce(const vector<int>& elecStates){
+   int mpiRank = MolDS_mpi::MpiProcess::GetInstance()->GetRank();
+   int mpiSize = MolDS_mpi::MpiProcess::GetInstance()->GetSize();
    this->CheckMatrixForce(elecStates);
    if(this->RequiresExcitedStatesForce(elecStates)){
       this->CalcEtaMatrixForce(elecStates);
       this->CalcZMatrixForce(elecStates);
    }
-   stringstream ompErrors;
+
+   // this loop is MPI-parallelized
+   for(int a=0; a<this->molecule->GetNumberAtoms(); a++){
+      if(a%mpiSize != mpiRank){continue;}
+      const Atom& atomA = *molecule->GetAtom(a);
+      int firstAOIndexA = atomA.GetFirstAOIndex();
+      int lastAOIndexA  = atomA.GetLastAOIndex();
+      stringstream ompErrors;
 #pragma omp parallel
-   {
-      double***** diatomicTwoElecTwoCore1stDerivs = NULL;
-      double***   diatomicOverlapAOs1stDerivs = NULL;
-      try{
-         this->MallocTempMatricesCalcForce(&diatomicOverlapAOs1stDerivs, &diatomicTwoElecTwoCore1stDerivs);
+      {
+         double***** diatomicTwoElecTwoCore1stDerivs = NULL;
+         double***   diatomicOverlapAOs1stDerivs = NULL;
+         try{
+            this->MallocTempMatricesCalcForce(&diatomicOverlapAOs1stDerivs, &diatomicTwoElecTwoCore1stDerivs);
+
 #pragma omp for schedule(auto)
-         for(int a=0; a<this->molecule->GetNumberAtoms(); a++){
-            const Atom& atomA = *molecule->GetAtom(a);
-            int firstAOIndexA = atomA.GetFirstAOIndex();
-            int lastAOIndexA  = atomA.GetLastAOIndex();
             for(int b=0; b<this->molecule->GetNumberAtoms(); b++){
                if(a == b){continue;}
                const Atom& atomB = *molecule->GetAtom(b);
@@ -2393,19 +2399,25 @@ void Mndo::CalcForce(const vector<int>& elecStates){
                      }
                   }
                } // end of excited state force
-            }    // end of for(int b)
-         }       // end of for(int a)
-      }          // end of try
-      catch(MolDSException ex){
+            }    // end of for(int b) with omp parallelization
+
+         }          // end of try for omp-for
+         catch(MolDSException ex){
 #pragma omp critical
-         ex.Serialize(ompErrors);
+            ex.Serialize(ompErrors);
+         }
+         this->FreeTempMatricesCalcForce(&diatomicOverlapAOs1stDerivs, &diatomicTwoElecTwoCore1stDerivs);
+      } // end of omp-parallelized region
+      // Exception throwing for omp-region
+      if(!ompErrors.str().empty()){
+         throw MolDSException::Deserialize(ompErrors);
       }
-      this->FreeTempMatricesCalcForce(&diatomicOverlapAOs1stDerivs, &diatomicTwoElecTwoCore1stDerivs);
-   }
-   // Exception throwing for omp-region
-   if(!ompErrors.str().empty()){
-      throw MolDSException::Deserialize(ompErrors);
-   }
+   }// end of for(int a) with MPI parallelization
+
+   // communication to reduce thsi->matrixForce on all node (namely, all_reduce)
+   int numTransported = elecStates.size()*this->molecule->GetNumberAtoms()*CartesianType_end;
+   MolDS_mpi::MpiProcess::GetInstance()->AllReduce(&this->matrixForce[0][0][0], numTransported, std::plus<double>());
+   
 }
 
 void Mndo::MallocTempMatricesCalcForce(double**** diatomicOverlapAOs1stDerivs, double****** diatomicTwoElecTwoCore1stDerivs) const{

--- a/src/zindo/ZindoS.cpp
+++ b/src/zindo/ZindoS.cpp
@@ -3731,7 +3731,7 @@ void ZindoS::CalcForce(const vector<int>& elecStates){
                                                     OrbitalType_end,
                                                     OrbitalType_end,
                                                     CartesianType_end);
-      } //end of parallelized region
+      } //end of omp-parallelized region
       // Exception throwing for omp-region
       if(!ompErrors.str().empty()){
          throw MolDSException::Deserialize(ompErrors);