wxmpimsgtmpl.h

Go to the documentation of this file.

#ifndef wxmpimsgtmpl_h
#define wxmpimsgtmpl_h
 
// WarpX includes
#include "wxmpitraits.h"
#include "wxmsgtmpl.h"
 
// std includes
#include <iostream>
 
struct WxMpiMsgStatus_v : public WxMsgStatus_v
{
    WxMpiMsgStatus_v() : data(0)
    {
    }
    MPI_Request request; // send/recv request
    void* data;          // pointer to memory for recv
};
 
template<typename T> class WxMpiMsgTmpl : public WxMsgTmpl<T>
{
public:
    WxMpiMsgTmpl(MPI_Comm comm) : _comm(comm), _sending(false)
    {
    }
 
    void send(unsigned num, const T* arr, unsigned recvRank, int tag) const override;
 
    void finishSend(WxMsgStatus ms) const override;
 
    WxMsgStatus
    startSend(unsigned num, const T* arr, unsigned recvRank, int tag) const override;
 
    void
    recv(unsigned num, std::vector<T>& array, unsigned sendRank, int tag) const override;
 
    void recv(unsigned num, T* array, unsigned sendRank, int tag) const override;
 
    WxMsgStatus startRecv(unsigned num, unsigned sendRank, int tag) const override;
 
    WxMsgStatus
    startRecv(unsigned num, T* arr, unsigned sendRank, int tag) const override;
 
    void allReduce(unsigned num, T* sendBuff, T* recvBuff, MPI_Op op) const override;
 
    void broadcast(unsigned num, T* buffer, int root) const override;
 
private:
    MPI_Comm _comm;
    bool _sending;
    MPI_Request _sendRequest;
};
 
extern template class WxMpiMsgTmpl<char>;
extern template class WxMpiMsgTmpl<unsigned char>;
extern template class WxMpiMsgTmpl<short>;
extern template class WxMpiMsgTmpl<unsigned short>;
extern template class WxMpiMsgTmpl<int>;
extern template class WxMpiMsgTmpl<unsigned int>;
extern template class WxMpiMsgTmpl<long>;
extern template class WxMpiMsgTmpl<unsigned long>;
extern template class WxMpiMsgTmpl<float>;
extern template class WxMpiMsgTmpl<double>;
extern template class WxMpiMsgTmpl<long double>;
extern template class WxMpiMsgTmpl<long long int>;
 
#endif //  wxmpimsgtmpl_h