Teuchos_MPISession.cpp
Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029 #include "Teuchos_MPISession.hpp"
00030 #include "Teuchos_TestForException.hpp"
00031
00032 using namespace Teuchos;
00033
00034 int MPISession::rank_ = 0 ;
00035 int MPISession::nProc_ = 1 ;
00036
00037 void MPISession::init(int* argc, void*** argv)
00038 {
00039 #ifdef HAVE_MPI
00040
00041 int mpiHasBeenStarted = 0;
00042 MPI_Initialized(& mpiHasBeenStarted);
00043 int mpierr = 0 ;
00044 if (!mpiHasBeenStarted)
00045 {
00046 mpierr = ::MPI_Init (argc, (char ***) argv);
00047 TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00048 "Error code=" << mpierr
00049 << " detected in MPI_Init()");
00050 }
00051
00052
00053 mpierr = ::MPI_Comm_rank (MPI_COMM_WORLD, &rank_);
00054 TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00055 "Error code=" << mpierr
00056 << " detected in MPI_Comm_rank()");
00057
00058
00059 mpierr = ::MPI_Comm_size (MPI_COMM_WORLD, &nProc_);
00060
00061 TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00062 "Error code=" << mpierr
00063 << " detected in MPI_Comm_size()");
00064
00065
00066 int nameLen;
00067 char procName[MPI_MAX_PROCESSOR_NAME];
00068 mpierr = ::MPI_Get_processor_name(procName,&nameLen);
00069
00070 TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00071 "Error code=" << mpierr
00072 << " detected in MPI_Get_processor_name()");
00073
00074 if (showStartupMessage())
00075 {
00076 std::cerr << "Teuchos::MPISession::init() started processor "
00077 << procName << std::endl;
00078 }
00079 else
00080 {
00081 #else
00082 std::cerr << "Teuchos::MPISession::init() started serial run" << std::endl;
00083 #endif
00084 #ifdef HAVE_MPI
00085 }
00086 #endif
00087 }
00088
00089 void MPISession::finalize()
00090 {
00091 #ifdef HAVE_MPI
00092 int mpierr = ::MPI_Finalize();
00093
00094 TEST_FOR_EXCEPTION(mpierr != 0, std::runtime_error,
00095 "Error code=" << mpierr << " detected in MPI_Finalize()");
00096 #endif
00097 }