Question

我正在尝试编写MPI发送/接收代码，但我遇到了问题。我想查看接收消息。如果我有来自任何来源的消息，请阅读此内容。否则，继续编码。

MPI_Comm_size(MPI_COMM_WORLD,&p);
MPI_Comm_rank(MPI_COMM_WORLD,&id);
MPI_Status status;
int source,ask,request,answer,tag=10;
for(i=first; i<last; i++){
    if(there is a message for my id){
        MPI_Recv(&request,1,MPI_INT,MPI_ANY_SOURCE,tag,MPI_COMM_WORLD,&status);
        answer = request*request;
        source = status.MPI_SOURCE;
        MPI_Send(&answer,1,MPI_INT,source,tag+1,MPI_COMM_WORLD);
    }
    square=i*i;
    if(last < square){
    MPI_Send(&square,1,MPI_INT,id-1,tag,MPI_COMM_WORLD);
    MPI_Recv(&square,1,MPI_INT,id-1,tag+1,MPI_COMM_WORLD,&status);
    }
    local[i]=square;
}

如果我无法使用if语句检查接收消息，是否可以检查超时或等待表达式？ *我知道这段代码是微不足道的，但我希望它可以帮助我们。

由于

Answer 1

很难确切地告诉您想要什么，但听起来您可能正在寻找的是与测试配对的非阻止接收或与{MPI_IMPROBE配对的MPI_MRECV 1}}。

因此，对于非阻塞接收版本，您的代码最终会更像：

int flag = 0;
for (i=first; i<last; i++) {
    MPI_Irecv(&data, 1, MPI_INT, MPI_ANY_SOURCE, tag, MPI_COMM_WORLD, &mpi_request);
    MPI_Test(&mpi_request, &flag, &status);
    while (!flag) {
         // Do other stuff
         MPI_Test(&mpi_request, &flag, &status);
    }
    // Do stuff with message
}

带探针的版本会略有不同，但最后非常相似：

MPI_Message message;
int flag;
for (i=first; i<last; i++) {
    MPI_Improbe(MPI_ANY_SOURCE, tag, MPI_COMM_WORLD, &flag, &message, &status);
    while (!flag) {
         // Do other stuff
         MPI_Test(&mpi_request, &flag, &status);
    }
    MPI_Mrecv(&data, 1, MPI_INT, &message, &status);
    // Do stuff with message
}

如何在MPI中检查接收消息

1 个答案: