mpi_wrapper.hpp

Go to the documentation of this file.

/*
  mptensor - Parallel Library for Tensor Network Methods
 
  Copyright 2016 Satoshi Morita
 
  mptensor is free software: you can redistribute it and/or modify it
  under the terms of the GNU Lesser General Public License as
  published by the Free Software Foundation, either version 3 of the
  License, or (at your option) any later version.
 
  mptensor is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  Lesser General Public License for more details.
 
  You should have received a copy of the GNU Lesser General Public
  License along with mptensor.  If not, see
  <https://www.gnu.org/licenses/>.
*/
 
#ifndef _MPI_WRAPPER_HPP_
#define _MPI_WRAPPER_HPP_
 
#ifndef _NO_MPI
 
#include <mpi.h>
#include <vector>
 
namespace mptensor {
 
namespace mpi_wrapper {
 
 
template <typename C>
C allreduce_sum(C val, const MPI_Comm &comm);
 
 
template <typename C>
std::vector<C> allreduce_vec(const std::vector<C> &vec, const MPI_Comm &comm);
 
 
template <typename C>
void send_recv_vector(const std::vector<C> &send_vec, int dest, int sendtag,
                      std::vector<C> &recv_vec, int source, int recvtag,
                      const MPI_Comm &comm, MPI_Status &status);
 
 
template <typename C>
void alltoallv(const C *sendbuf, const int *sendcounts, const int *sdispls,
               C *recvbuf, const int *recvcounts, const int *rdispls,
               const MPI_Comm &comm);
 
 
template <typename C>
void bcast(C *buffer, int count, int root, const MPI_Comm &comm);
 
}  // namespace mpi_wrapper
}  // namespace mptensor
 
#endif  // _NO_MPI
#endif  // _MPI_WRAPPER_HPP_