MPI发送矩阵列(C ++)

时间:2015-01-20 16:59:55

标签: c++ matrix mpi send

我尝试将具有等级1和2的过程的矩阵的第二和第三列与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;
}

1 个答案:

答案 0 :(得分:-1)

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

rbuf必须指向9 int的数组。这就是您按照here解释的Abort trap: 6错误的原因。您正试图通过matrix[0][3]

撰写matrix[0][8]

此外,您的recv_type必须为MPI_INT