MPI_IBcastW.hpp

/*
 * MPI_IBcastW.hpp
 *
 *  Created on: Apr 8, 2017
 *      Author: i-bird
 *      Modified by: Abhinav Singh
 */
 
#ifndef OPENFPM_VCLUSTER_SRC_MPI_WRAPPER_MPI_IBCASTW_HPP_
#define OPENFPM_VCLUSTER_SRC_MPI_WRAPPER_MPI_IBCASTW_HPP_
 
 
 
#include <mpi.h>
 
class MPI_IBcastWB
{
public:
    static inline void bcast(size_t proc ,void * buf, size_t sz, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(buf,sz,MPI_BYTE, proc , ext_comm,&req));
    }
};
 
template<typename T> class MPI_IBcastW
{
public:
    template<typename Memory>  static inline void bcast(size_t proc ,openfpm::vector<T, Memory> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size() * sizeof(T),MPI_BYTE, proc , ext_comm,&req));
    }
};
 
 
template<> class MPI_IBcastW<int>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<int> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_INT, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<unsigned int>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<unsigned int> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_UNSIGNED, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<short>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<short> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_SHORT, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<unsigned short>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<unsigned short> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_UNSIGNED_SHORT, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<char>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<char> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_CHAR, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<unsigned char>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<unsigned char> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_UNSIGNED_CHAR, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<size_t>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<size_t> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_UNSIGNED_LONG, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<long int>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<long int> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_LONG, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<float>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<float> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_FLOAT, proc , ext_comm,&req));
    }
};
 
template<> class MPI_IBcastW<double>
{
public:
    static inline void bcast(size_t proc ,openfpm::vector<double> & v, MPI_Request & req, MPI_Comm ext_comm)
    {
        MPI_SAFE_CALL(MPI_Ibcast(v.getPointer(), v.size(),MPI_DOUBLE, proc , ext_comm,&req));
    }
};
 
 
template<typename vect>
struct bcast_inte_impl
{
    MPI_Comm ext_comm;
 
    vect & send;
 
    openfpm::vector<MPI_Request> & req;
 
    size_t root;
 
    inline bcast_inte_impl(vect & send,
                           openfpm::vector<MPI_Request> & req,
                           size_t root, MPI_Comm ext_comm=MPI_COMM_WORLD)
    :send(send),req(req),root(root), ext_comm(ext_comm)
    {};
 
    template<typename T>
    inline void operator()(T& t)
    {
        typedef typename boost::mpl::at<typename vect::value_type::type,T>::type send_type;
 
        // Create one request
        req.add();
 
        // gather
        MPI_IBcastWB::bcast(root,&send.template get<T::value>(0),send.size()*sizeof(send_type),req.last(), ext_comm);
    }
};
 
template<bool is_lin_or_inte>
struct b_cast_helper
{
     template<typename T, typename Mem, template<typename> class layout_base >
 static void bcast_(openfpm::vector<MPI_Request> & req,
                        openfpm::vector<T,Mem,layout_base> & v,
                        size_t root, MPI_Comm ext_comm)
    {
        // Create one request
        req.add();
 
        // gather
        MPI_IBcastW<T>::bcast(root,v,req.last(), ext_comm);
    }
};
 
template<>
struct b_cast_helper<false>
{
     template<typename T, typename Mem, template<typename> class layout_base >
     static void bcast_(openfpm::vector<MPI_Request> & req,
                        openfpm::vector<T,Mem,layout_base> & v,
                        size_t root, MPI_Comm ext_comm)
    {
         bcast_inte_impl<openfpm::vector<T,Mem,layout_base>> bc(v,req,root,ext_comm);
 
         boost::mpl::for_each_ref<boost::mpl::range_c<int,0,T::max_prop>>(bc);
    }
};
 
#endif /* OPENFPM_VCLUSTER_SRC_MPI_WRAPPER_MPI_IBCASTW_HPP_ */