75 lines
2.3 KiB
C++
75 lines
2.3 KiB
C++
// @HEADER
|
|
// ***********************************************************************
|
|
//
|
|
// Teuchos: Common Tools Package
|
|
// Copyright (2004) Sandia Corporation
|
|
//
|
|
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
|
|
// license for use of this work by or on behalf of the U.S. Government.
|
|
//
|
|
// 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.1 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
|
|
// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
|
|
//
|
|
// ***********************************************************************
|
|
// @HEADER
|
|
|
|
#ifndef TEUCHOS_MPISESSION_H
|
|
#define TEUCHOS_MPISESSION_H
|
|
|
|
/*! \file Teuchos_MPISession.hpp
|
|
\brief A MPI utilities class, providing methods for initializing,
|
|
finalizing, and querying the global MPI session
|
|
*/
|
|
#include "Teuchos_ConfigDefs.hpp"
|
|
|
|
#ifdef HAVE_MPI
|
|
#include "mpi.h"
|
|
#endif
|
|
|
|
namespace Teuchos
|
|
{
|
|
/**
|
|
* \brief This class provides methods for initializing, finalizing,
|
|
* and querying the global MPI session.
|
|
*/
|
|
class MPISession
|
|
{
|
|
public:
|
|
//! Initializer, calls MPI_Init() if necessary
|
|
static void init(int* argc, void*** argv);
|
|
|
|
//! Returns the process rank relative to MPI_COMM_WORLD
|
|
static int getRank() {return rank_;}
|
|
|
|
//! Returns the number of processors in MPI_COMM_WORLD
|
|
static int getNProc() {return nProc_;}
|
|
|
|
//! Finalizer, calls MPI_Finalize() if necessary
|
|
static void finalize();
|
|
|
|
/** Set to true if a message should be written by each processor
|
|
* at startup. */
|
|
static bool& showStartupMessage() {static bool rtn=false; return rtn;}
|
|
|
|
private:
|
|
|
|
static int rank_;
|
|
static int nProc_;
|
|
};
|
|
}
|
|
|
|
#endif
|