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