registered_types.h

Go to the documentation of this file.

#ifndef SOPT_MPI_TYPES_H
#define SOPT_MPI_TYPES_H
 
#include "sopt/config.h"
#ifdef SOPT_MPI
#include <complex>
#include <mpi.h>
namespace sopt::mpi {
using MPIType = decltype(MPI_CHAR);
// Some MPI libraries don't actually have a type defined that will work in the above line
// so the line below can be used instead
// using MPIType = int;
 
template <typename T>
struct Type;
 
static_assert(not std::is_same<char, std::int8_t>::value, "");
#define SOPT_MACRO(TYPE)        \
  template <>                   \
  struct Type<TYPE> {           \
    static const MPIType value; \
  };
SOPT_MACRO(std::int8_t);
SOPT_MACRO(std::int16_t);
SOPT_MACRO(std::int32_t);
SOPT_MACRO(std::int64_t);
SOPT_MACRO(std::uint8_t);
SOPT_MACRO(std::uint16_t);
SOPT_MACRO(std::uint32_t);
SOPT_MACRO(std::uint64_t);
 
#ifndef SOPT_CHAR_ARCH
SOPT_MACRO(char);
#endif
#ifndef SOPT_LONG_ARCH
SOPT_MACRO(signed long);
#endif
#ifndef SOPT_ULONG_ARCH
SOPT_MACRO(unsigned long);
#endif
 
SOPT_MACRO(float);
SOPT_MACRO(double);
SOPT_MACRO(long double);
SOPT_MACRO(std::complex<float>);
SOPT_MACRO(std::complex<double>);
SOPT_MACRO(std::complex<long double>);
#undef SOPT_MACRO
 
template <typename T>
inline constexpr MPIType registered_type(T const &) {
  return Type<T>::value;
}
 
namespace details {
template <typename... Ts>
struct make_void {
  using type = void;
};
template <typename... Ts>
using void_t = typename make_void<Ts...>::type;
}  // namespace details
template <typename T, typename = details::void_t<>>
class is_registered_type : public std::false_type {};
template <typename T>
class is_registered_type<T, details::void_t<decltype(Type<T>::value)>> : public std::true_type {};
} // namespace sopt::mpi
#endif
#endif /* ifndef SOPT_TYPES */