MPI_Bcast一个动态的2d结构数组

时间:2011-11-01 18:37:04

标签: c mpi

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

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;
 }

现在我得到输入并将其存储在矩阵中并调用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);

1 个答案:

答案 0 :(得分:0)

这似乎出现了很多:如果你在C中传递带有MPI的多维数组,你必须将多维数组视为一个数组数组。我想在你的第一个例子中你很幸运矩阵[0] [0](由于连续的mallocs?)位于大块内存的开头(因此没有seg错误)。

在第二个(工作)示例中,您已经描述了内存地址(不仅是内存区域的开头(matrix [0] [0])),还描述了所有内部指针。