mpi_type_traits.hh

Go to the documentation of this file.

/*
    @@@@@@@@  @@           @@@@@@   @@@@@@@@ @@
   /@@/////  /@@          @@////@@ @@////// /@@
   /@@       /@@  @@@@@  @@    // /@@       /@@
   /@@@@@@@  /@@ @@///@@/@@       /@@@@@@@@@/@@
   /@@////   /@@/@@@@@@@/@@       ////////@@/@@
   /@@       /@@/@@//// //@@    @@       /@@/@@
   /@@       @@@//@@@@@@ //@@@@@@  @@@@@@@@ /@@
   //       ///  //////   //////  ////////  //

   Copyright (c) 2016, Triad National Security, LLC
   All rights reserved.
                                                                              */
#pragma once

#include <flecsi-config.h>

#if !defined(FLECSI_ENABLE_MPI)
#error FLECSI_ENABLE_MPI not defined! This file depends on MPI!
#endif

#include <mpi.h>

namespace flecsi {
namespace util {

template<typename TYPE>
struct mpi_typetraits {

  inline static MPI_Datatype type() {
    static MPI_Datatype data_type = MPI_DATATYPE_NULL;

    if(data_type == MPI_DATATYPE_NULL) {
      MPI_Type_contiguous(sizeof(TYPE), MPI_BYTE, &data_type);
      MPI_Type_commit(&data_type);
    }
    return data_type;
  }
};

template<>
struct mpi_typetraits<size_t> {
  inline static MPI_Datatype type() {
    if(sizeof(size_t) == 8) {
      return MPI_UNSIGNED_LONG_LONG;
    }
    else {
      return MPI_UNSIGNED;
    } // if
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<char> {
  inline static MPI_Datatype type() {
    return MPI_SIGNED_CHAR;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<unsigned char> {
  inline static MPI_Datatype type() {
    return MPI_UNSIGNED_CHAR;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<short> {
  inline static MPI_Datatype type() {
    return MPI_SHORT;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<unsigned short> {
  inline static MPI_Datatype type() {
    return MPI_UNSIGNED_SHORT;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<int> {
  inline static MPI_Datatype type() {
    return MPI_INT;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<unsigned> {
  inline static MPI_Datatype type() {
    return MPI_UNSIGNED;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<long> {
  inline static MPI_Datatype type() {
    return MPI_LONG;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<double> {
  inline static MPI_Datatype type() {
    return MPI_DOUBLE;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<float> {
  inline static MPI_Datatype type() {
    return MPI_FLOAT;
  }
}; // mpi_typetraits

template<>
struct mpi_typetraits<long double> {
  inline static MPI_Datatype type() {
    return MPI_LONG_DOUBLE;
  }
}; // mpi_typetraits

} // namespace util
} // namespace flecsi