Homec4science
Diffusion µSpectre (feat/add_lam_pixels) Browse

muSpectre/src/libmufft/communicator.hh
fa590e5a9f93feat/add_lam_pixels

Last Change
Raw File

communicator.hh
View Options

/**
* @file communicator.hh
*
* @author Lars Pastewka <lars.pastewka@imtek.uni-freiburg.de>
*
* @date 07 Mar 2018
*
* @brief abstraction layer for the distributed memory communicator object
*
* Copyright © 2017 Till Junge
*
* µFFT 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, or (at
* your option) any later version.
*
* µFFT 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
* General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with µFFT; see the file COPYING. If not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* * Boston, MA 02111-1307, USA.
*
* Additional permission under GNU GPL version 3 section 7
*
* If you modify this Program, or any covered work, by linking or combining it
* with proprietary FFT implementations or numerical libraries, containing parts
* covered by the terms of those libraries' licenses, the licensors of this
* Program grant you additional permission to convey the resulting work.
*/
#ifndef SRC_LIBMUFFT_COMMUNICATOR_HH_
#define SRC_LIBMUFFT_COMMUNICATOR_HH_
#ifdef WITH_MPI
#include <mpi.h>
#endif
namespace muFFT {
#ifdef WITH_MPI
template <typename T>
decltype(auto) mpi_type() {}
template <>
inline decltype(auto) mpi_type<char>() {
return MPI_CHAR;
}
template <>
inline decltype(auto) mpi_type<short>() { // NOLINT
return MPI_SHORT;
}
template <>
inline decltype(auto) mpi_type<int>() {
return MPI_INT;
}
template <>
inline decltype(auto) mpi_type<long>() { // NOLINT
return MPI_LONG;
}
template <>
inline decltype(auto) mpi_type<unsigned char>() {
return MPI_UNSIGNED_CHAR;
}
template <>
inline decltype(auto) mpi_type<unsigned short>() { // NOLINT
return MPI_UNSIGNED_SHORT;
}
template <>
inline decltype(auto) mpi_type<unsigned int>() {
return MPI_UNSIGNED;
}
template <>
inline decltype(auto) mpi_type<unsigned long>() { // NOLINT
return MPI_UNSIGNED_LONG;
}
template <>
inline decltype(auto) mpi_type<float>() {
return MPI_FLOAT;
}
template <>
inline decltype(auto) mpi_type<double>() {
return MPI_DOUBLE;
}
//! lightweight abstraction for the MPI communicator object
class Communicator {
public:
using MPI_Comm_ref = std::remove_pointer_t<MPI_Comm> &;
explicit Communicator(MPI_Comm comm = MPI_COMM_NULL) : comm{*comm} {};
~Communicator() {}
//! get rank of present process
int rank() const {
if (&comm == MPI_COMM_NULL)
return 0;
int res;
MPI_Comm_rank(&this->comm, &res);
return res;
}
//! get total number of processes
int size() const {
if (&comm == MPI_COMM_NULL)
return 1;
int res;
MPI_Comm_size(&this->comm, &res);
return res;
}
//! sum reduction on scalar types
template <typename T>
T sum(const T & arg) const {
if (&comm == MPI_COMM_NULL)
return arg;
T res;
MPI_Allreduce(&arg, &res, 1, mpi_type<T>(), MPI_SUM, &this->comm);
return res;
}
MPI_Comm get_mpi_comm() { return &this->comm; }
private:
MPI_Comm_ref comm;
};
#else /* WITH_MPI */
//! stub communicator object that doesn't communicate anything
class Communicator {
public:
Communicator() {}
~Communicator() {}
//! get rank of present process
int rank() const { return 0; }
//! get total number of processes
int size() const { return 1; }
//! sum reduction on scalar types
template <typename T>
T sum(const T & arg) const {
return arg;
}
};
#endif
} // namespace muFFT
#endif // SRC_LIBMUFFT_COMMUNICATOR_HH_
c4science · Help