Teuchos_GlobalMPISession.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                    Teuchos: Common Tools Package
//
// Copyright 2004 NTESS and the Teuchos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef TEUCHOS_GLOBAL_MPI_SESSION_HPP
#define TEUCHOS_GLOBAL_MPI_SESSION_HPP
 
#include "TeuchosCore_ConfigDefs.hpp"
 
#include "Teuchos_ArrayView.hpp"
 
 
namespace Teuchos {
 
//
class TEUCHOSCORE_LIB_DLL_EXPORT GlobalMPISession
{
public:
 
 
 
  GlobalMPISession( int* argc, char*** argv, std::ostream *out = &std::cout );
 
  ~GlobalMPISession();
 
 
 
 
  static void abort();
 
 
 
  static bool mpiIsInitialized();
 
  static bool mpiIsFinalized();
 
  static int getRank();
 
  static int getNProc();
 
  static void barrier();
 
  static int sum(int localVal);
 
  static void allGather(int localVal, const ArrayView<int> &allVals);
 
#ifdef HAVE_TEUCHOSCORE_KOKKOS
  static std::vector<std::string> getArgv ();
#endif // HAVE_TEUCHOSCORE_KOKKOS
 
private:
 
  static bool haveMPIState_;
  static bool mpiIsFinalized_;
  static int rank_;
  static int nProc_;
#ifdef HAVE_TEUCHOSCORE_KOKKOS
  static std::vector<std::string> argvCopy_;
#endif // HAVE_TEUCHOSCORE_KOKKOS
 
  static void initialize( std::ostream *out );
 
  static void justInTimeInitialize();
 
};
 
} // namespace Teuchos
 
#endif // TEUCHOS_GLOBAL_MPI_SESSION_HPP