在MPI中如何创建结构体结构并将其发送到多个进程

时间:2014-11-12 21:14:10

标签: c struct mpi

所以我有这个结构

typedef struct{
     float x;
     float y;
     int centroid;
}point;

我也有这个结构

typedef struct{
     int csize;//the current size
     int tsize;//the total size
     point * data;//the data carried
}ArrayList;

这些第二个结构在C中形成一个动态增长的数组(实现动态增长的函数并且正常工作)。 我究竟如何创建一个结构并在c中使用MPI发送它? 我查看了其他帖子,如struct serialization in C and transfer over MPI和其他帖子,但我找不到解决问题的方法。

我很感激一些帮助。

最佳, 盖多

编辑 - 可能的重复不能解决我的问题,我的问题是一个包含指向动态增长的结构数组的指针的结构。重复的问题涉及在结构中有一个本机类型数组。

1 个答案:

答案 0 :(得分:2)

手动将结构数组序列化到缓冲区并发送缓冲区是一个糟糕的主意,因为它在发送方和接收方都引入了另一个内存副本。

发送MPI结构数组与任何其他对象的数组没有什么不同;你只需要创建一个结构类型 - 有many examples hereelsewhere - 确保计算偏移量和大小,因为你的结构中可能会插入填充。然后只传递一个数组:

#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include "mpi.h"

typedef struct{
    float x;
    float y;
    int centroid;
} point;

typedef struct{
    int csize;//the current size
    int tsize;//the total size
    point * data;//the data carried
} ArrayList;


int main(int argc, char **argv)
{
    MPI_Init(&argc, &argv);

    int rank, size;
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);
    MPI_Comm_size(MPI_COMM_WORLD, &size);

    assert( size >= 2 );
    const int npts_avail=20;
    point points[npts_avail];

    ArrayList list;
    list.data = points;

    if (rank == 0) {
        int npts_used=10;
        list.csize = npts_used;
        list.tsize = npts_avail;
        for (int i=0; i<list.csize; i++) {
            points[i].x = 1.*i;
            points[i].y = -2.*i;
            points[i].centroid = i;
        }
    }

    const int nfields=3;
    MPI_Aint disps[nfields];
    int blocklens[] = {1,1,1};
    MPI_Datatype types[] = {MPI_FLOAT, MPI_FLOAT, MPI_INT};

    disps[0] = offsetof( point, x );
    disps[1] = offsetof( point, y );
    disps[2] = offsetof( point, centroid );

    MPI_Datatype istruct, pstruct;
    MPI_Type_create_struct(nfields, blocklens, disps, types, &istruct );
    MPI_Type_create_resized( istruct, 0, (char *)&(points[1]) - (char *)(&points[0]), &pstruct );
    MPI_Type_commit(&pstruct);

    if (rank == 0) {
        MPI_Send( &(list.csize), 1, MPI_INT, 1, 0, MPI_COMM_WORLD);
        MPI_Send( &(list.tsize), 1, MPI_INT, 1, 0, MPI_COMM_WORLD);
        MPI_Send( list.data, list.csize, pstruct, 1, 0, MPI_COMM_WORLD);
    } else if (rank == 1) {
        MPI_Recv( &(list.csize), 1, MPI_INT, 0, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
        MPI_Recv( &(list.tsize), 1, MPI_INT, 0, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
        MPI_Recv( list.data, list.csize, pstruct, 0, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
    }

    if (rank == 1) {
        printf("Received: \n");
        for (int i=0; i<list.csize; i++) {
            printf(" (%f, %f): %d\n", points[i].x, points[i].y, points[i].centroid);
        }
    }

    MPI_Finalize();
}

Running给出了预期的输出:

$ mpirun -np 2 ./structs
Received:
 (0.000000, -0.000000): 0
 (1.000000, -2.000000): 1
 (2.000000, -4.000000): 2
 (3.000000, -6.000000): 3
 (4.000000, -8.000000): 4
 (5.000000, -10.000000): 5
 (6.000000, -12.000000): 6
 (7.000000, -14.000000): 7
 (8.000000, -16.000000): 8
 (9.000000, -18.000000): 9

请注意,您也可以构造一个ArrayList的MPI结构,并使用它 - 除了每次重新发送(在发送方)可能必须更改数据的位移,并且在接收方它可以&甚至可以确定,直到您知道需要接收的数据量为止。所以最好首先发送大小(在两条消息中,就像我在这里,或者更好的是,将它作为一个两个整数的消息发送),然后发送结构数组,如图所示。