mpi.h

#ifndef PROFILER_SRC_MOLPRO_PROFILER_MPI_H_
#define PROFILER_SRC_MOLPRO_PROFILER_MPI_H_
 
#include <cstdint>
 
#ifndef HAVE_MPI_H
#if defined(OMPI_MPI_H) || defined(MPI_INCLUDED)
#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>
#endif
 
#ifdef HAVE_PPIDD_H
#include <ppidd.h>
#endif
 
namespace molpro {
namespace profiler {
 
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
}
 
inline MPI_Comm comm_global() {
#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__
  {
    int64_t size;
    PPIDD_Size(&size);
    if (size > 0)
      return MPI_Comm_f2c(PPIDD_Worker_comm());
  }
#else
#ifdef GA_H
  if (GA_MPI_Comm() != NULL && 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;
}
 
int init();
int finalize();
 
} // namespace profiler
} // namespace molpro
 
#endif // PROFILER_SRC_MOLPRO_PROFILER_MPI_H_