MPI Sendrecv与MPI_ANY_SOURCE

时间:2015-04-06 21:14:05

标签: c mpi

是否可以进行MPI_Sendrecv交换,其中一方不知道另一方的排名?如果没有,那么最好的方法是什么(我的下一个猜测只是一对发送和recvs)?

例如在C中,如果我想在等级0和其他等级之间交换整数,这种类型的东西会起作用吗?:

MPI_Status stat;
if(rank){
  int someval = 0;
  MPI_Sendrecv(&someval, 1, MPI_INT, 0, 1, &recvbuf, 1, MPI_INT, 0, MPI_ANY_TAG, MPI_COMM_WORLD, &stat);
}else{
  int someotherval = 1;
  MPI_Sendrecv(&someotherval, 1, MPI_INT, MPI_ANY_SOURCE, someotherval, &recvbuf, 1, MPI_INT, MPI_ANY_SOURCE, MPI_ANY_TAG, MPI_COMM_WORLD, &stat);
}

编辑: 看起来不可能。我把以下内容作为一种包装器来添加我需要的功能。

void slave_sendrecv(const void *sendbuf, int sendcount, MPI_Datatype sendtype,
    int dest, int sendtag, void *recvbuf, int recvcount,
    MPI_Datatype recvtype, int source, int recvtag, MPI_Status *status){

    MPI_Send(sendbuf, sendcount, sendtype, dest, sendtag, MPI_COMM_WORLD);
    MPI_Recv(recvbuf, recvcount, recvtype, source, recvtag, MPI_COMM_WORLD, status);
}

void anon_sendrecv(const void *sendbuf, int sendcount, MPI_Datatype sendtype,
    int sendtag, void *recvbuf, int recvcount,
    MPI_Datatype recvtype, int recvtag, MPI_Status *status){

    int anon_rank;
    MPI_Recv(recvbuf, recvcount, recvtype, MPI_ANY_SOURCE, recvtag, MPI_COMM_WORLD, status);
    anon_rank = status -> MPI_SOURCE;
    MPI_Send(sendbuf, sendcount, sendtype, anon_rank, sendtag, MPI_COMM_WORLD);
}

编辑2:根据Patrick的回答,看起来上面的slave_sendrecv函数看起来不需要,你可以在结尾处使用常规的MPI_Sendrecv,知道它发送给谁。

1 个答案:

答案 0 :(得分:3)

简短回答:不。

该标准不允许在任何发送程序中使用MPI_ANY_SOURCE作为目标排名dest。这是有道理的,因为您无法在不知道目的地的情况下发送消息。

但是,该标准允许您将MPI_Sendrecv与常规MPI_Send / MPI_Recv配对:

  

通过常规接收操作可以接收由发送 - 接收操作发送的消息   或通过探测操作探测;发送 - 接收操作可以接收发送的消息   通过定期发送操作。

在您的情况下,流程0必须先收到,然后回答:

MPI_Status stat;
if(rank){
  int someval = 0;
  MPI_Sendrecv(&someval, 1, MPI_INT, 0, 1, &recvbuf, 1, MPI_INT, 0, MPI_ANY_TAG, MPI_COMM_WORLD, &stat);
}else{
  int someotherval = 1;
  MPI_Recv(&recvbuf, 1, MPI_INT, MPI_ANY_SOURCE, MPI_ANY_TAG, MPI_COMM_WORLD, &stat);
  // answer to process `stat.MPI_SOURCE` using `someotherval` as tag
  MPI_Send(&someotherval, 1, MPI_INT, stat.MPI_SOURCE, someotherval, MPI_COMM_WORLD);
}