MPI 发送矩阵列 (C++)

MPI send matrix columns (C++)

我尝试将矩阵的第二列和第三列从具有 MPI 的秩一和二的进程发送到秩为零的进程。

在网上找到了这个例子http://www.mcs.anl.gov/research/projects/mpi/mpi-standard/mpi-report-1.1/node70.htm#Figure4,写了一段代码:

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

using namespace std;

int main(int argc, char *argv[]) {
    int id;
    int matrix[3][3];
    int matrixB[9];

    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &id);

    for(int i=0; i<3; i++)
        for(int j=0; j<3; j++)
            if(id == 0)
                matrix[i][j] = 0;
            else
                matrix[i][j] = j+1;

    MPI_Datatype matrixSpalte;

    MPI_Type_vector(3, 1, 3, MPI_INT, &matrixSpalte);
    MPI_Type_commit(&matrixSpalte);

    MPI_Gather(&matrix[0][id], 1, matrixSpalte, &matrixB[0], 1, matrixSpalte, 0, MPI_COMM_WORLD);

    if(id == 0)
        for(int i=0; i<9; i++) {
            if(i % 3 == 0)          
                cout << endl;

            cout << matrixB[i] << " (" << i << ") ";

        }


    MPI_Finalize();
    return 0;
}

但输出是:

0 (0) 0 (1) 1351992736 (2)
0 (3) 254423040 (4) 1 (5)
0 (6) 2 (7) 1351992752 (8)

应该是:

1 (0) 2 (1) 3 (2)
1 (3) 2 (4) 3 (5)
1 (6) 2 (7) 3 (8)

而且我不知道为什么这不起作用。我希望有人能帮我找出我代码中的错误。

您真诚的,

亨氏

解决方案(感谢 Jonathan Dursi):

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

using namespace std;

int main(int argc, char *argv[]) {
    int id;
    int matrix[3][3];
    int matrixB[9];

    MPI_Init(&argc, &argv);
    MPI_Comm_rank(MPI_COMM_WORLD, &id);

    for(int i=0; i<3; i++)
        for(int j=0; j<3; j++)
            if(id == 0)
                matrix[i][j] = 0;
            else
                matrix[i][j] = j+1;

    MPI_Datatype matrixSpalte, tmp;

    MPI_Type_vector(3, 1, 3, MPI_INT, &tmp);
    MPI_Type_create_resized(tmp, 0, sizeof(int), &matrixSpalte);  // !!!
    MPI_Type_commit(&matrixSpalte);

    MPI_Gather(&matrix[0][id], 1, matrixSpalte, matrixB, 1, matrixSpalte, 0, MPI_COMM_WORLD);

    if(id == 0)
        for(int i=0; i<9; i++) {
            if(i % 3 == 0)          
                cout << endl;

            cout << matrixB[i] << " (" << i << ") ";

        }


    MPI_Finalize();
    return 0;
}

MPI_Gather 应该传递一个缓冲区,该缓冲区可以将所有元素存储在一个连续的数组中。看看the picture here

也就是rbuf必须指向9个int的数组。这就是您收到 Abort trap: 6 错误的原因,如 here 所述。您正在尝试通过 matrix[0][8]

写入 matrix[0][3]

此外,您的 recv_type 必须是 MPI_INT