fullDuplexCommBuffer.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2023 - 2025 NeoN authors
//
// SPDX-License-Identifier: MIT
 
#pragma once
 
#include <vector>
#include <string>
 
#include "NeoN/core/mpi/environment.hpp"
#include "NeoN/core/mpi/halfDuplexCommBuffer.hpp"
#include "NeoN/core/view.hpp"
 
namespace NeoN
{
 
#ifdef NF_WITH_MPI_SUPPORT
 
namespace mpi
{
 
class FullDuplexCommBuffer
{
public:
 
    FullDuplexCommBuffer() = default;
 
    FullDuplexCommBuffer(
        MPIEnvironment mpiEnviron,
        std::vector<std::size_t> sendSize,
        std::vector<std::size_t> receiveSize
    )
        : send_(mpiEnviron, sendSize), receive_(mpiEnviron, receiveSize) {};
 
    inline bool isCommInit() const { return send_.isCommInit() && receive_.isCommInit(); }
 
    template<typename valueType>
    void initComm(std::string commName)
    {
        send_.initComm<valueType>(commName);
        receive_.initComm<valueType>(commName);
    }
 
    template<typename valueType>
    View<valueType> getSend(const size_t rank)
    {
        return send_.get<valueType>(rank);
    }
 
    template<typename valueType>
    View<const valueType> getSend(const size_t rank) const
    {
        return send_.get<valueType>(rank);
    }
 
    template<typename valueType>
    View<valueType> getReceive(const size_t rank)
    {
        return receive_.get<valueType>(rank);
    }
 
    template<typename valueType>
    View<const valueType> getReceive(const size_t rank) const
    {
        return receive_.get<valueType>(rank);
    }
 
    inline void startComm()
    {
        send_.send();
        receive_.receive();
    }
 
    inline bool isComplete() { return send_.isComplete() && receive_.isComplete(); }
 
    inline void waitComplete()
    {
        send_.waitComplete();
        receive_.waitComplete();
    }
 
    inline void finaliseComm()
    {
        send_.finaliseComm();
        receive_.finaliseComm();
    }
 
private:
 
    HalfDuplexCommBuffer send_;    
    HalfDuplexCommBuffer receive_; 
};
 
} // namespace mpi
 
#endif
 
}