c++ - "Deadlock"与 MPI_Waitall

Question

我正在尝试将一些数据从每个进程发送到它的两个邻居(按等级)。为此，我尝试使用非阻塞发送和接收调用 MPI_Isend/MPI_Irecv 以及对 MPI_Waitall 的相应调用，以确保传输在使用数据之前发生。

下面是我正在尝试做的一个最小示例

#include <mpi.h>
#include <iostream>

using namespace std;

int main(int argc, char* argv[])
{
    MPI_Init(&argc,&argv);
    int rank,comm_size;

    MPI_Comm_rank(MPI_COMM_WORLD,&rank);
    MPI_Comm_size(MPI_COMM_WORLD,&comm_size);

    MPI_Request reqs[4];

    int sbuff1[10];
    int sbuff2[10];
    int rbuff1[10];
    int rbuff2[10];

    int count = 0;

    if(rank>0)
    {
        MPI_Irecv(rbuff1,10,MPI_INT,rank-1,0,MPI_COMM_WORLD,&reqs[count]);
        count++;
        MPI_Isend(sbuff1,10,MPI_INT,rank-1,1,MPI_COMM_WORLD,&reqs[count]);
        count++;
    }

    if(rank<comm_size-1)
    {
        MPI_Irecv(rbuff2,10,MPI_INT,rank+1,2,MPI_COMM_WORLD,&reqs[count]);
        count++;
        MPI_Isend(sbuff2,10,MPI_INT,rank+1,3,MPI_COMM_WORLD,&reqs[count]);
        count++;
    }

    MPI_Waitall(count,reqs,MPI_STATUS_IGNORE);

    if(rank==0)
            cout << "Successfully exchanged data." << endl;

    MPI_Finalize();

    return 0;
}

不幸的是，这段代码永远无法打印出最后的消息。

然后我的问题是:数据发送是否正确，但我对 MPI_Waitall 的调用不知何故出错了？如果是这样，我需要更改什么才能使这个简单的代码运行？如果不是，还有什么问题？

Answer 1

您设置的发送/接收标签永远不会匹配。

这个小修复

#include <mpi.h>
#include <iostream>

using namespace std;

int main(int argc, char* argv[])
{
    MPI_Init(&argc,&argv);
    int rank,comm_size;

    MPI_Comm_rank(MPI_COMM_WORLD,&rank);
    MPI_Comm_size(MPI_COMM_WORLD,&comm_size);

    MPI_Request reqs[4];

    int sbuff1[10];
    int sbuff2[10];
    int rbuff1[10];
    int rbuff2[10];

    int count = 0;

    if(rank>0)
    {

        MPI_Irecv(rbuff1,10,MPI_INT,rank-1,0,MPI_COMM_WORLD,&reqs[count]);
        count++;
        MPI_Isend(sbuff1,10,MPI_INT,rank-1,1,MPI_COMM_WORLD,&reqs[count]);
        count++;
    }

    if(rank<comm_size-1)
    {
        /// changed tag 2 -> 1 
        // MPI_Irecv(rbuff2,10,MPI_INT,rank+1,2,MPI_COMM_WORLD,&reqs[count]);
        MPI_Irecv(rbuff2,10,MPI_INT,rank+1,1,MPI_COMM_WORLD,&reqs[count]);
        count++;

        /// changed tag 3 -> 0 
        // MPI_Isend(sbuff2,10,MPI_INT,rank+1,3,MPI_COMM_WORLD,&reqs[count]);
        MPI_Isend(sbuff2,10,MPI_INT,rank+1,0,MPI_COMM_WORLD,&reqs[count]);
        count++;
    }

    MPI_Waitall(count,reqs,MPI_STATUS_IGNORE);

    if(rank==0)
            cout << "Successfully exchanged data." << endl;

    MPI_Finalize();

    return 0;
}

让它工作

$> mpirun -n 5 ./testling 
Successfully exchanged data.

使用 mpich2 v1.5 (Hydra) 在 x86_84 Linux 上运行