如何在mpi中传输数据

How transfer data in mpi?

本文关键字：传输 数据 mpi      更新时间：2023-10-16

正如我所知，在mpi中，我们可以在计算机之间传输一些数组。我们可以转移其他东西而不是数组吗？例如，是否可以直接传输"集合"数据？

是的，您可以使用MPI传输任意复杂类型的数据。所有通信操作都采用数据类型参数。这可能是与C/C++中的内置类型（例如MPI_INT/int或MPI_DOUBLE/double）或更复杂的用户定义类型相对应的基本预定义数据类型。

我最近刚刚发布了一个完整的工作（纯C）示例，其中integer值后面跟着一个doubles数组，在MPI_Type_create_struct的帮助下传输，它可以用来定义这样一个用户定义的数据类型：

#include <mpi.h>
#include <stdio.h>
int main(int argc, char *argv[]) {
    MPI_Init(&argc, &argv);
    int i, rank, tag = 1;
    MPI_Status status;
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    // Array of doubles plus element count
    typedef struct {
        int row;
        double elements[4];
    } My_array;
    // Derived datatype for an array of doubles plus element count
    MPI_Datatype MY_ARRAY_TYPE;
    const int nr_blocks = 2;
    int blocklengths[2] = {1, 4};
    MPI_Datatype oldtypes[2] = {MPI_INT, MPI_DOUBLE};
    MPI_Aint displacements[2];
    displacements[0] = offsetof(My_array, row);
    displacements[1] = offsetof(My_array, elements);
    MPI_Type_create_struct(nr_blocks, blocklengths, displacements,
               oldtypes, &MY_ARRAY_TYPE);
    MPI_Type_commit(&MY_ARRAY_TYPE);
    if(rank == 0) {
        My_array array1  = {3, 3.1, 3.2, 3.3, 3.4};
        MPI_Send(&array1, 1, MY_ARRAY_TYPE, 1, tag, MPI_COMM_WORLD);
    }
    if(rank == 1) {
        My_array array2;
        MPI_Recv(&array2, 1, MY_ARRAY_TYPE, 0, tag, MPI_COMM_WORLD, &status);
        printf("Rank %d received elements of row %d:n", rank, array2.row);
        for(i = 0; i < 4; i++)
            printf("t%.1fn", array2.elements[i]);
    }
    MPI_Type_free(&MY_ARRAY_TYPE);
    MPI_Finalize();
}

还有其他不太通用的派生数据类型构造函数更易于使用；你可以在这个答案中看到两个例子。

相关文章：