iteration_payload.hpp

#ifndef __ITERATION_PAYLOAD_HPP__
#define __ITERATION_PAYLOAD_HPP__
 
#include <cma_utils/alias.hpp>
#include <mpi.h>
#include <span>
#include <vector>
 
namespace WrapMPI
{

  class IterationPayload
  {
  public:
    std::vector<double> liquid_volumes;             // n_c
    std::vector<std::size_t> liquid_neighbors_flat; // m*n_c
    std::vector<double> proba_leaving_flat;         // m*n_c
    std::vector<double> liquid_out_flows;           // n_c
 
    /*
      Estimated payload size with
      n_c = 500
      m =6
      size = 56000 bytes ≈ 54.69 KiB ≈ 0.055 MB > Eager limit MPI should
      triggers SM
    */

    explicit IterationPayload(size_t volumes);

    bool recv(size_t source, MPI_Status* status) noexcept;
  };

  class HostIterationPayload
  {
  public:
    void fill(const CmaUtils::IterationStatePtrType& current_reactor_state);
 
    [[nodiscard]] bool sendAll(std::size_t n_rank) noexcept;
 
    void wait() noexcept;
#ifdef NDEBUG // FIXME Needed for unite test
  private:
#endif
    std::span<const double> liquid_volumes;
    std::span<const std::size_t> liquid_neighbors_flat;
    std::span<const double> proba_leaving_flat;
    std::span<const double> liquid_out_flows;
 
    bool to_wait;

    [[nodiscard]] bool send(size_t rank) noexcept;
 
    static constexpr std::size_t n_vector_send = 4;
    std::array<MPI_Request, n_vector_send> requests;
  };
 
} // namespace WrapMPI
 
#endif //__ITERATION_PAYLOAD_HPP__