cd4b27992a
git-svn-id: svn://scm.gforge.inria.fr/svnroot/paradiseo@94 331e1502-861f-0410-8da2-ba01fb791d7f
73 lines
2.1 KiB
C++
73 lines
2.1 KiB
C++
// -*- mode: c++; c-indent-level: 4; c++-member-init-indent: 8; comment-column: 35; -*-
|
|
|
|
// "node.cpp"
|
|
|
|
// (c) OPAC Team, LIFL, August 2005
|
|
|
|
/* This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2 of the License, or (at your option) any later version.
|
|
|
|
This library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with this library; if not, write to the Free Software
|
|
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
Contact: paradiseo-help@lists.gforge.inria.fr
|
|
*/
|
|
|
|
#include <mpi.h>
|
|
#include <vector>
|
|
#include <map>
|
|
#include <string>
|
|
#include <cassert>
|
|
|
|
static int rk, sz; /* Rank & size */
|
|
|
|
static std :: map <std :: string, int> name_to_rk;
|
|
|
|
static std :: vector <std :: string> rk_to_name;
|
|
|
|
int getNodeRank () {
|
|
|
|
return rk;
|
|
}
|
|
|
|
int getNumberOfNodes () {
|
|
|
|
return sz;
|
|
}
|
|
|
|
int getRankFromName (const std :: string & __name) {
|
|
|
|
return atoi (__name.c_str ());
|
|
}
|
|
|
|
void initNode (int * __argc, char * * * __argv) {
|
|
|
|
int provided;
|
|
MPI_Init_thread (__argc, __argv, MPI_THREAD_FUNNELED, & provided);
|
|
assert (provided == MPI_THREAD_FUNNELED); /* The MPI implementation must be multi-threaded.
|
|
Yet, only one thread performs the comm.
|
|
operations */
|
|
MPI_Comm_rank (MPI_COMM_WORLD, & rk); /* Who ? */
|
|
MPI_Comm_size (MPI_COMM_WORLD, & sz); /* How many ? */
|
|
|
|
char names [sz] [MPI_MAX_PROCESSOR_NAME];
|
|
int len;
|
|
|
|
/* Processor names */
|
|
MPI_Get_processor_name (names [0], & len); /* Me */
|
|
MPI_Allgather (names, MPI_MAX_PROCESSOR_NAME, MPI_CHAR, names, MPI_MAX_PROCESSOR_NAME, MPI_CHAR, MPI_COMM_WORLD); /* Broadcast */
|
|
|
|
for (int i = 0; i < sz; i ++) {
|
|
rk_to_name.push_back (names [i]);
|
|
name_to_rk [names [i]] = i;
|
|
}
|
|
}
|
|
|