如何在mpi中传输数据?

时间:2015-06-29 13:28:09

标签: c++ set mpi

正如我在mpi中所知,我们可以在计算机之间传输一些数组。我们可以转移任何其他东西而不是数组吗? 例如,是否可以转移"设置"数据直接?

1 个答案:

答案 0 :(得分:0)

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

我刚刚posted一个完整的工作(纯C)示例,其中integer值后跟一个double数组,在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%.1f\n", array2.elements[i]);
    }
    MPI_Type_free(&MY_ARRAY_TYPE);
    MPI_Finalize();
}

还有其他更简单的派生数据类型构造函数更易于使用;你可以在this answer中看到两个例子。