2

我目前正在做一个项目，我需要使用 openmpi 实现并行 fft 算法。我有一段正在编译的代码，但是当我在集群上运行它时，我遇到了分段错误。

我对哪里出了问题有预感，但我认为我对指针和引用的理解还不够，无法进行有效的修复。

第一个可能出错的块是将数组传递给辅助函数。我相信要么我的循环不一致，要么我不明白如何传递这些指针并取回我需要的东西。

第二个可能的位置是在实际的 mpi_Send/Recv 命令中。我正在发送 openmpi c 数据类型不支持的类型，因此我使用 mpi_byte 类型来发送原始数据。这是一个可行的选择吗？或者我应该寻找这种方法的替代方法。

/* function declarations */
double complex get_block(double complex c[], int start, int stop);

double complex put_block(double complex from[], double complex to[], 
            int start, int stop);

void main(int argc, char **argv)
{
  /* Initialize MPI */
  MPI_Init(&argc, &argv);

  double complex c[N/p];
  int myid;
  MPI_Comm_rank(MPI_COMM_WORLD, &myid);
  //printf("My id is %d\n",myid);

  MPI_Status status;

  int i;
  for(i=0;i<N/p;i++){
    c[i] = 1.0 + 1.0*I;
  }

  int j = log(p)/log(2) + 1;
  double q;
  double complex z;
  double complex w = exp(-2*PI*I/N);
  double complex block[N/(2*p)]; // half the size of chunk c
  int e,l,t,k,m,rank,plus,minus;
  int temp = (log(N)-log(p))/log(2);
  //printf("temp = %d", temp);

  for(e = 0; e < (log(p)/log(2)); e++){
    /* loop constants */
    t = pow(2,e); l = pow(2,e+temp);
    q = n/2*l; z = cpow(w,(complex)q); 
    j = j-1; int v = pow(2,j);

    if(e != 0){
      plus = (myid + p/v)%p;
      minus = (myid - p/v)%p;
    } else {
      plus = myid + p/v;
      minus = myid - p/v;
    }

    if(myid%t == myid%(2*t)){
      MPI_Recv((char*)&c, 
           sizeof(c),
           MPI_BYTE,
           plus,
           MPI_ANY_TAG,
           MPI_COMM_WORLD,
           &status);

      /* transform */
      for(k = 0; k < N/p; k++){
    m = (myid * N/p + k)%l;
    c[k] = c[k] + c[k+N/v] * cpow(z,m);
    c[k+N/v] = c[k] - c[k + N/v] * cpow(z,m); 
    printf("(k,k+N/v) = (%d,%d)\n",k,k+N/v);
    }*/
      printf("\n\n");
      /* end transform */

      *block = get_block(c, N/v, N/v + N/p + 1);
      MPI_Send((char*)&block,
           sizeof(block),
           MPI_BYTE, 
           plus,
           1, 
           MPI_COMM_WORLD);
    } else {
      // send data of this PE to the (i- p/v)th PE
      MPI_Send((char*)&c,
           sizeof(c),
           MPI_BYTE,
           minus,
           1, 
           MPI_COMM_WORLD);
      // after the transformation, receive data from (i-p/v)th PE
      //      and store them in c:
      MPI_Recv((char*)&block,
           sizeof(block),
           MPI_BYTE,
           minus, 
           MPI_ANY_TAG, 
           MPI_COMM_WORLD,
              &status);

      *c = put_block(block, c, N/v, N/v + N/p - 1);
      //printf("Process %d send/receive %d\n",myid, plus);
    }
  }
  /* shut down MPI */
  MPI_Finalize();
}

/* helper functions */
double complex get_block(double complex *c, int start, int stop)
{
  double complex block[stop - start + 1];
  //printf("%d = %d\n",sizeof(block)/sizeof(double complex), sizeof(&c)/sizeof(double  complex));
  int j = 0;
  int i;
  for(i = start; i < stop+1; i++){
    block[j] = c[i];
    j = j+1;
  }
  return *block;
}


double complex put_block(double complex from[], double complex to[], int start, int stop)
{  
  int j = 0;
  int i;
  for(i = start; i<stop+1; i++){
    to[i] = from[j];
    j = j+1;
  }
  return *to;
}

我非常感谢您的反馈！

4

2 回答 2

1

您以错误的方式使用数组/指向数组的指针。例如，您将数组声明为双复数块 [N]，这很好（虽然不常见，但在大多数情况下最好使用 malloc），然后您通过 MPI_Recv(&block) 接收到它。但是“block”已经是指向该数组的指针，因此通过编写“&block”，您将指针的指针传递给 MPI_Recv。这不是它所期望的。如果你想使用“&”符号，你必须写 &block[0]，它会给你指向块数组第一个元素的指针。

于 2012-06-19T17:22:48.130 回答
1

你试过调试你的代码吗？这在并行设置中可能会很痛苦，但它可以准确地告诉您失败的位置，通常还会告诉您失败的原因。

如果您使用的是 Linux 或 OS X，则可以在命令行上按如下方式运行代码：

mpirun -np 4 xterm -e gdb -ex run --args ./yourprog yourargs

我假设yourprog是您的程序的名称，并且yourargs是您要传递的任何命令行参数。

该命令将启动四个xterm窗口。每个都xterm将依次gdb按照 option 的指定启动-e。然后将执行选项指定gdb的命令，并使用给定的选项启动可执行文件，如.run-ex--args

你得到的是四个xterm与 MPI 并行运行程序的四个实例的窗口。如果任何实例崩溃，gdb将告诉您在哪里以及为什么。

于 2012-06-19T16:44:53.753 回答