MPI_Bcast 结构的动态二维数组答案

【问题标题】：MPI_Bcast a dynamic 2d array of structsMPI_Bcast 结构的动态二维数组
【发布时间】：2011-11-01 18:37:04
【问题描述】：

我从这里 (MPI_Bcast a dynamic 2d array) 跟随 Jonathan 的代码到 MPI_Bcast 一个动态分配的二维结构数组。结构如下：

typedef struct {
   char num[MAX];
   int neg;
} bignum;

由于这不是本机数据类型，我使用 MPI_Type_create_struct 提交此数据类型。我省略了这里的代码，因为它适用于我正在做的另一个项目。

我用来动态创建（“连续？”）数组的方法是调用：

 bignum **matrix = creatematrix(DIMMAX,DIMMAX);

具体实现为：

 bignum ** creatematrix (int num_rows, int num_cols){
   int i;
   bignum *one_d = (bignum*)malloc (num_rows * num_cols * sizeof (bignum));
   bignum **two_d = (bignum**)malloc (num_rows * sizeof (bignum *));

   for (i = 0; i < num_rows; i++)
   two_d[i] = &(one_d[i*num_cols]);//one_d + (i * num_cols);

   return two_d;
 }

现在我将获取输入并将其存储在 matrix 中，并将 MPI_Bcast 调用为：

MPI_Bcast(&(matrix[0][0]), DIMMAX*DIMMAX, Bignum, 0, MPI_COMM_WORLD);

似乎没有分段错误，但问题是只广播了第一行（matrix[0]）。根旁边的所有其他等级仅具有第一行的数据。其他行保持不变。

看起来不仅仅是分配一个连续的内存块。是不是我错过了什么导致广播不成功？

编辑：

我摸索了一个奇怪的解决方法，使用嵌套结构。谁能解释为什么这种方法有效而上面的方法无效？

typedef struct {
   char num[MAX];
   int neg;
} bignum;

typedef struct {
   bignum elements[DIMMAX];
} row;

int main(/*int argc,char *argv[]*/){
  int dim, my_rank, comm_sz;
  int i,j,k; //iterators
  bignum global_sum = {{0},0};
  row *matrix = (row*)malloc(sizeof(row)*DIMMAX);
  row *tempmatrix = (row*)malloc(sizeof(row)*DIMMAX);
  MPI_Init(NULL,NULL);
  MPI_Comm_size(MPI_COMM_WORLD, &comm_sz);
  MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);

  //CREATE DERIVED DATATYPE Bignum
  MPI_Datatype Bignum;  
  MPI_Datatype type[4] = {MPI_LB, MPI_CHAR, MPI_INT, MPI_UB};
  int blocklen[4] = {1,MAX,1,1};
  MPI_Aint disp[4];
  //get offsets 
  MPI_Get_address(&global_sum, disp);
  MPI_Get_address(&global_sum.num, disp+1); 
  MPI_Get_address(&global_sum.neg, disp+2);
  //MPI_Get_address(&local_sum+1, disp+3);
  int base;
  base = disp[0];
  for (i=0;i<3;i++){
      disp[i] -= base;  
  }
  disp[3] = disp[2]+4;  //int
  MPI_Type_create_struct(4,blocklen,disp,type,&Bignum);
  MPI_Type_commit(&Bignum);

  //CREATE DERIVED DATATYPE BignumRow
  MPI_Datatype BignumRow;   
  MPI_Datatype type2[3] = {MPI_LB, Bignum, MPI_UB};
  int blocklen2[3] = {1,DIMMAX,1};  
  MPI_Aint disp2[3] = {0,0,DIMMAX*(MAX+4)};
  MPI_Type_create_struct(3,blocklen2,disp2,type2,&BignumRow);
  MPI_Type_commit(&BignumRow);

  MPI_Bcast(&dim, 1, MPI_INTEGER, 0, MPI_COMM_WORLD);
  MPI_Bcast(matrix, dim, BignumRow, 0, MPI_COMM_WORLD);

【问题讨论】：

标签： c mpi

【解决方案1】：

这似乎出现了很多：如果您在 C 中使用 MPI 传递多维数组，您必须将多维数组视为数组的数组。我认为在您的第一个示例中，您很幸运 matrix[0][0]（由于连续的 malloc？）位于大内存块的开头（因此，没有 seg 错误）。

在第二个（工作）示例中，您不仅描述了内存区域（matrix[0][0]）开头的内存地址，还描述了所有内部指针的内存地址。

【讨论】：