如何使用C在MPI中使用MPI_Datatype发送嵌套结构

时间:2011-05-11 19:56:41

标签: c distributed mpi distributed-computing

我正在尝试使用MPI_Datatype发送以下结构,但MPI_Send在发送结构时崩溃。我想知道如何处理这种情况。这是我编写的用于定义新MPI数据类型的代码:

   typedef struct
    {
       double x;
       double y;
    } vertex;

   typedef struct
   {
        int num_vertices;
        vertex vertex[2];
   } vertex_list;


       MPI_Datatype vertexType;
        MPI_Type_contiguous(2,MPI_DOUBLE,&vertexType);
        MPI_Type_commit(&vertexType);

      MPI_Datatype vertexListType;
        MPI_Datatype typev[3] = {MPI_INT, vertexType, MPI_UB};
        int blocklenv[3] = {1, 2, 1};
        MPI_Aint dispv[3];
        /* compute displacements of structure components */
        MPI_Address( vertexl, dispv);
        MPI_Address( vertexl[0].vertex, dispv+1);
        MPI_Address(  vertexl+1, dispv+2);
        base = dispv[0];

        for (i=0; i <3; i++)
         dispv[i] -= base;

        /* build datatype describing structure */
        MPI_Type_struct( 3, blocklenv, dispv, typev, &vertexListType);
        MPI_Type_commit(&vertexListType);

https://docs.google.com/document/d/1OQFtx0ClkKQx7X91BlVgiizs5D9jShhtgsKafrgC7hk/edit?hl=en

1 个答案:

答案 0 :(得分:3)

我不确定你的结构是否会按照你想要的方式工作,但我可以分享使用MPI_Send发送结构的经验。

不是创建一个显式的MPI数据类型,而是可以简单地发送结构本身,因为它的所有内容都在一块连续的内存中。诀窍是为MPI_Send操作提供正确的大小和数据类型。

使用你的结构,这是我过去所做的(假设已经定义了变量vertex_list list):

MPI_Send(&list, sizeof(vertex_list), MPI_BYTE, <rankToSendTo>, <tagInteger>, <comm>);

因此要发送的数据缓冲区是指向列表结构的指针,缓冲区的大小是以字节为单位的vertex_list的大小,MPI数据类型只是字节。

在接收端,您只需提供一个vertex_list引用作为接收缓冲区:

vertex_list receiveList;
MPI_Recv(&receiveList, sizeof(vertex_list), MPI_BYTE, <rankOfSender>, <tagInteger>, <comm>, <status>);

希望有所帮助!