Parallel.hpp

/*------------------------------------------------------------------------*/
/*                 Copyright 2010 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.  */
/*  Export of this program may require a license from the                 */
/*  United States Government.                                             */
/*------------------------------------------------------------------------*/

#ifndef stk_util_parallel_Parallel_hpp
#define stk_util_parallel_Parallel_hpp

// stk_config.h resides in the build directory and contains the
// complete set of #define macros for build-dependent features.

#include <stk_util/stk_config.h>

//----------------------------------------------------------------------
// Parallel machine

#if defined( STK_HAS_MPI )

#include <mpi.h>

namespace stk_classic {



typedef MPI_Comm     ParallelMachine ;


typedef MPI_Datatype ParallelDatatype ;

inline ParallelMachine parallel_machine_null() { return MPI_COMM_NULL ; }

inline ParallelMachine parallel_machine_init( int * argc , char *** argv )
{
  MPI_Init( argc , argv );
  return MPI_COMM_WORLD ;
}

inline void parallel_machine_finalize()
{
  MPI_Finalize();
}

}

//----------------------------------------
// Other parallel communication machines go here
// as '#elif defined( STK_HAS_<name> )'

//----------------------------------------
// Stub for non-parallel

#else

// Some needed stubs
#define MPI_Comm int
#define MPI_COMM_WORLD 0
#define MPI_COMM_SELF 0
#define MPI_Barrier( a ) (void)a

namespace stk_classic {

typedef int ParallelMachine ;
typedef int ParallelDatatype ;

inline ParallelMachine parallel_machine_null() { return 0 ; }

inline ParallelMachine parallel_machine_init( int * , char *** )
{ return 0 ; }

inline void parallel_machine_finalize()
{}

}

#endif

//----------------------------------------------------------------------
// Common parallel machine needs.

namespace stk_classic {

unsigned parallel_machine_size( ParallelMachine parallel_machine );

unsigned parallel_machine_rank( ParallelMachine parallel_machine );

void parallel_machine_barrier( ParallelMachine parallel_machine);
}

//----------------------------------------------------------------------

#endif