mpi.h

Go to the documentation of this file.

#ifndef UTILITIES_SRC_MOLPRO_MPI_H_
#define UTILITIES_SRC_MOLPRO_MPI_H_
 
#include <cstdint>
 
#ifndef HAVE_MPI_H
#if defined(OMPI_MPI_H) || defined(MPI_INCLUDED)
#define HAVE_MPI_H
#endif
#if defined __has_include
#if __has_include(<mpi.h>)
#endif
#define HAVE_MPI_H
#endif
#endif
 
#ifdef HAVE_MPI_H
#include <mpi.h>
#else
#define MPI_Comm_c2f(x) x
using MPI_Comm = int64_t;
#define MPI_COMM_NULL 0
#define MPI_WIN_NULL 0
#define MPI_INFO_NULL 0
using MPI_Win = int64_t;
inline void _MPI_nullfunction(const void* x) {}
#define MPI_Barrier(x) _MPI_nullfunction(&x)
#define MPI_Win_free(x) _MPI_nullfunction(x)
#define MPI_Win_allocate(x1,x2,x3,x4,x5,x6) _MPI_nullfunction(x5)
#define MPI_Win_lock(x1,x2,x3,x4) _MPI_nullfunction(&x4)
#define MPI_Win_unlock(x1,x2) _MPI_nullfunction(&x2)
#endif
 
#ifdef HAVE_GA_H
#include <ga.h>
#include <ga-mpi.h>
#endif
 
#ifdef HAVE_PPIDD_H
#include <ppidd.h>
#endif
 
namespace molpro {
namespace mpi {
 
inline MPI_Comm comm_self() {
#ifdef HAVE_MPI_H
  int flag;
  MPI_Initialized(&flag);
  if (!flag)
    return MPI_COMM_NULL;
  return MPI_COMM_SELF;
#else
  return 0;
#endif
}
 
#ifdef MOLPRO
extern "C" int molpro_mppx();
#endif
 
inline MPI_Comm comm_global() {
#ifdef MOLPRO
  if (molpro_mppx()) return comm_self();
#endif
#ifdef HAVE_MPI_H
  int flag;
  MPI_Initialized(&flag);
  if (!flag) {
    MPI_Init(0, nullptr);
    return MPI_COMM_WORLD;
  }
#else
  return comm_self();
#endif
#ifdef __PPIDD_H__
  {
    if (PPIDD_Size() > 0)
      return MPI_Comm_f2c(PPIDD_Worker_comm());
  }
#else
#ifdef HAVE_GA_H
  if (GA_MPI_Comm() != 0 && GA_MPI_Comm() != MPI_COMM_NULL) {
    return GA_MPI_Comm();
  }
#endif
#endif
#ifdef HAVE_MPI_H
  return MPI_COMM_WORLD;
#endif
  return comm_self();
}
 
inline int size_global() {
  int size = 1;
#ifdef HAVE_MPI_H
  MPI_Comm_size(comm_global(), &size);
#endif
  return size;
}
 
inline int rank_global() {
  int rank = 0;
#ifdef HAVE_MPI_H
  MPI_Comm_rank(comm_global(), &rank);
#endif
  return rank;
}
 
inline int init() {
#ifdef HAVE_MPI_H
  return MPI_Init(0, nullptr);
#else
  return 0;
#endif
}
 
inline int finalize() {
#ifdef HAVE_MPI_H
  return MPI_Finalize();
#else
  return 0;
#endif
}
 
 
} // namespace mpi
} // namespace molpro
 
extern "C" inline int64_t mpicomm_global() { return (int64_t)MPI_Comm_c2f(molpro::mpi::comm_global()); }
 
extern "C" inline int64_t mpicomm_self() { return (int64_t)MPI_Comm_c2f(molpro::mpi::comm_self()); }
 
extern "C" inline int64_t mpisize_global() { return molpro::mpi::size_global(); }
 
extern "C" inline int64_t mpirank_global() { return molpro::mpi::rank_global(); }
 
extern "C" inline int mpi_init() { return molpro::mpi::init(); }
 
extern "C" inline int mpi_finalize() { return molpro::mpi::finalize(); }
#endif // UTILITIES_SRC_MOLPRO_MPI_H_