nojhan/eodev
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Multiple roles demo has now 2 sub masters. Needs to be launched with at least 7 nodes.

Browse source
This commit is contained in:
Benjamin Bouvier 2012-06-22 17:56:08 +02:00
commit 5bf03dec2b
1 changed files with 29 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

44
eo/test/mpi/multipleRoles.cpp
View file

@ -13,34 +13,42 @@ using namespace std;
// 1 : worker of general job, master of subjob // 1 : worker of general job, master of subjob
// 2 and more : workers of subjob // 2 and more : workers of subjob
struct plusOne : public eoUF< int&, void > struct SubWork: public eoUF< int&, void >
{ {
void operator() ( int & x ) void operator() ( int & x )
{ {
cout << "Subjob is being applied." << endl; cout << "Subwork phase." << endl;
++x; ++x;
} }
}; };
void subtask( vector<int>& v ) void subtask( vector<int>& v, int rank )
{ {
DynamicAssignmentAlgorithm algo( 2, eo::REST_OF_THE_WORLD ); vector<int> workers;
plusOne plusOneInstance; workers.push_back( rank + 2 );
ParallelApply<int> job( plusOneInstance, v, algo, 1 ); workers.push_back( rank + 4 );
DynamicAssignmentAlgorithm algo( workers );
SubWork sw;
ParallelApply<int> job( sw, v, algo, rank );
job.run(); job.run();
} }
struct transmit : public eoUF< vector<int>&, void > struct Work: public eoUF< vector<int>&, void >
{ {
void operator() ( vector<int>& v ) void operator() ( vector<int>& v )
{ {
cout << "Into the master subjob..." << endl; cout << "Work phase..." << endl;
subtask( v ); subtask( v, MpiNode::comm().rank() );
for( int i = 0; i < v.size(); ++i )
{
v[i] *= 2;
}
} }
}; };
int main(int argc, char** argv) int main(int argc, char** argv)
{ {
// eo::log << eo::setlevel( eo::debug );
MpiNode::init( argc, argv ); MpiNode::init( argc, argv );
vector<int> v; vector<int> v;
@ -50,19 +58,19 @@ int main(int argc, char** argv)
v.push_back(7); v.push_back(7);
v.push_back(42); v.push_back(42);
transmit transmitInstance;
vector< vector<int> > metaV; vector< vector<int> > metaV;
metaV.push_back( v ); metaV.push_back( v );
metaV.push_back( v );
switch( MpiNode::comm().rank() ) switch( MpiNode::comm().rank() )
{ {
case 0: case 0:
case 1: case 1:
case 2:
{ {
// only one node is assigned to subjob mastering Work w;
DynamicAssignmentAlgorithm algo( 1 ); DynamicAssignmentAlgorithm algo( 1, 2 );
ParallelApply< vector<int> > job( transmitInstance, metaV, algo, 0 ); ParallelApply< vector<int> > job( w, metaV, algo, 0 );
job.run(); job.run();
if( job.isMaster() ) if( job.isMaster() )
{ {
@ -80,7 +88,13 @@ int main(int argc, char** argv)
default: default:
{ {
// all the other nodes are sub workers // all the other nodes are sub workers
subtask( v ); int rank = MpiNode::comm().rank();
if ( rank == 3 or rank == 5 )
{
subtask( v, 1 );
} else {
subtask( v, 2 );
}
} }
break; break;
} }