channel.cpp

Go to the documentation of this file.

 
#include "channel.h"
#include "layout.h"
#include <mpi-ext.h> // for Fujitsu extension
 
 
/*
       std::vector version with the allocator templated
       this class is a template independent part
                               [5 May 2022 I.Kanamori]
 */
 
//====================================================================
// class Channel_communicator
 
//====================================================================
 
int Channel_communicator::send_init(int count, int idir, int ipm, void *buf)
{
  LOG;
 
  assert(ipm == 1 || ipm == -1);
  assert(Communicator_impl::Layout::m_ndim <= max_dimension);
 
  //  m_buf.resize(count);
 
  int dest = (ipm == Forward) ?
             Communicator_impl::Layout::m_ipe_up[idir] : Communicator_impl::Layout::m_ipe_dn[idir];
  int dir2 = 2 * idir + ((ipm == Forward) ? 0 : 1);
  int tag  = max_dimension * send_num[dir2] + dir2 + tag_offset;
  send_num[dir2]++;
 
  int retv = FJMPI_Prequest_send_init(buf, sizeof(element_type) * count, MPI_BYTE, dest, tag, Communicator_impl::world(), &m_request);
 
  return retv;
}
 
 
//====================================================================
 
int Channel_communicator::recv_init(int count, int idir, int ipm, void *buf)
{
  LOG;
 
  assert(ipm == 1 || ipm == -1);
  assert(Communicator_impl::Layout::m_ndim <= max_dimension);
 
  //  m_buf.resize(count);
 
  int src = (ipm == Forward) ?
            Communicator_impl::Layout::m_ipe_up[idir] : Communicator_impl::Layout::m_ipe_dn[idir];
 
  int dir2 = 2 * idir + ((ipm == Forward) ? 1 : 0);
  int tag  = max_dimension * recv_num[dir2] + dir2 + tag_offset;
  recv_num[dir2]++;
 
  int retv = FJMPI_Prequest_recv_init(buf, sizeof(element_type) * count, MPI_BYTE, src, tag, Communicator_impl::world(), &m_request);
 
  return retv;
}
 
 
//====================================================================
Channel_communicator::Channel_communicator()
{
  LOG;
}
 
 
//====================================================================
Channel_communicator::~Channel_communicator()
{
  LOG;
  MPI_Request_free(&m_request);
}
 
 
//====================================================================
int Channel_communicator::start()
{
  LOG;
  return FJMPI_Prequest_start(&m_request);
}
 
 
//====================================================================
int Channel_communicator::wait()
{
  LOG;
  return MPI_Wait(&m_request, &m_status);
}
 
 
int Channel_communicator::send_num[8] = { 0, 0, 0, 0, 0, 0, 0, 0 };
int Channel_communicator::recv_num[8] = { 0, 0, 0, 0, 0, 0, 0, 0 };
 
//====================================================================
// class ChannelSet
ChannelSet::ChannelSet(int count)
  : m_array(count), m_status(count), m_nreq(0)
{
  LOG;
}
 
 
//====================================================================
int ChannelSet::append(const MPI_Request& r)
{
  LOG;
 
  if (m_nreq >= m_array.size()) {
    return MPI_ERR_BUFFER;
  }
  m_array[m_nreq++] = r;
 
  return MPI_SUCCESS;
}
 
 
//====================================================================
int ChannelSet::start()
{
  LOG;
  return FJMPI_Prequest_startall(m_nreq, &m_array[0]);
}
 
 
//====================================================================
int ChannelSet::wait()
{
  LOG;
  //  return MPI_Waitall(m_nreq, &m_array[0], (MPI_Status *)0);
  return MPI_Waitall(m_nreq, &m_array[0], &m_status[0]);
}
 
 
//====================================================================
// Channel with defaul allocator
template class Channel_impl<std::allocator<char> >;
 
//====================================================================
//============================================================END=====