二维数组的 MPI 数据类型

MPI data type for 2D array

本文关键字：数据类型 MPI 二维数组      更新时间：2023-10-16

几个小时以来，我试图通过 MPI 将 2D 数组发送到不同的线程。代码看起来像这样（我会留下评论来展示我已经尝试过的技巧）：

类型定义：

// Define datatype
//MPI_Type_contiguous(N, MPI_INT, &mpi_vector);
//MPI_Type_vector(N, N, N, mpi_vector, &mpi_matrix);
//MPI_Type_vector(N, N, N, MPI_INT, &mpi_matrix);
//MPI_Type_contiguous(N, MPI_INTEGER, &mpi_vector);
//MPI_Type_contiguous(N, mpi_vector, &mpi_matrix);
MPI_Type_vector(N, N, 0, MPI_INTEGER, &mpi_matrix);
//MPI_Type_commit(&mpi_vector);
MPI_Type_commit(&mpi_matrix);

发送和接收：

int** tmp = new int*[N];
switch(r) {
case T1:
    inputMatrix(tmp, 2);
    MPI_Send(tmp, 1, mpi_matrix, T2, 0, MPI_COMM_WORLD);
    //task_T1();
    break;
case T2:
    //task_T2();
    inputMatrix(tmp, -1);
    MPI_Recv(tmp, 1, mpi_matrix, T1, 0, MPI_COMM_WORLD, &s);
    for(int i = 0; i < N; i++)
    {
        for(int j = 0; j < N; j++) {
            cout << "  " << tmp[i][j];
        }
        cout << endl;
    }
    break;
}

我需要在早上（7 小时内）完成这项工作，我希望 smb 能够帮助我。

MPI 不支持单独分配每行的矩阵，即指向行的指针数组。MPI 仅适用于行连续存储在内存中的平面数组。使用平面数组：

int** tmp = ...

应该成为

int* tmp = new int[rows*cols];

然后tmp[row][col]应该作为tmp[row*cols + col]访问。

可以从单个行的MPI_INT构造连续类型，然后从整个矩阵的前一个连续类型构造连续类型。您还可以构造单个连续类型的rows*cols MPI_INT元素。第一种方法更灵活，因为您可能需要稍后发送单独的行。

相关文章：