zcy
/
openEMS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

MPI: allow requested number of splits

pull/1/head
Thorsten Liebig 2011-03-08 10:30:08 +01:00
parent eb7524b6f6
commit 8d09a7ee28
1 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
FDTD/openems_fdtd_mpi.cpp
View File

@ -112,13 +112,14 @@ bool openEMS_FDTD_MPI::SetupMPI(TiXmlElement* FDTD_Opts)
vector<unsigned int> SplitNumber[3];
string argNames[] = {"SplitPos_X", "SplitPos_Y", "SplitPos_Z"};
string arg_Pos_Names[] = {"SplitPos_X", "SplitPos_Y", "SplitPos_Z"};
string arg_N_Names[] = {"SplitN_X", "SplitN_Y", "SplitN_Z"};
const char* tmp = NULL;
for (int n=0;n<3;++n)
{
SplitNumber[n].push_back(0);
tmp = MPI_Elem->Attribute(argNames[n].c_str());
if (tmp)
tmp = MPI_Elem->Attribute(arg_Pos_Names[n].c_str());
if (tmp) //check if a split position is requested
{
vector<double> SplitLines = SplitString2Double(tmp, ',');
bool inside;
@ -130,6 +131,22 @@ bool openEMS_FDTD_MPI::SetupMPI(TiXmlElement* FDTD_Opts)
SplitNumber[n].push_back(line);
}
}
else //check if a number of splits is requested
{
int SplitN=0;
if (MPI_Elem->QueryIntAttribute( arg_N_Names[n].c_str(), &SplitN) == TIXML_SUCCESS)
{
if (SplitN<=1)
break;
vector<unsigned int> jobs = AssignJobs2Threads(m_Original_Grid->GetQtyLines(n)-1, SplitN, true);
unsigned int line=0;
for (size_t i = 0; i<jobs.size()-1;++i)
{
line += jobs.at(i);
SplitNumber[n].push_back(line);
}
}
}
SplitNumber[n].push_back(m_Original_Grid->GetQtyLines(n)-1);
unique(SplitNumber[n].begin(), SplitNumber[n].end());