Ошибка сегментации - MPI FFTW - PullRequest
0 голосов
/ 01 февраля 2019

У меня есть очень простой код 2D MPI FFTW следующим образом.

int main(int argc, char **argv){
  int N0 = 4, N1 = 4;
  fftw_plan plan;
  fftw_complex *data; //local data of course
  ptrdiff_t alloc_local, local_n0, local_0_start;

  MPI_Init(&argc, &argv);
  fftw_mpi_init();

  int id, p, ierr, count;
  ierr = MPI_Comm_size ( MPI_COMM_WORLD, &p );
  ierr = MPI_Comm_rank ( MPI_COMM_WORLD, &id );

  /* get local data size and allocate */
  alloc_local = fftw_mpi_local_size_2d(N0, N1, MPI_COMM_WORLD,
                       &local_n0, &local_0_start);

  data = fftw_alloc_complex(alloc_local);
  memset( data, 0, alloc_local*sizeof(fftw_complex));
  printf("Processor %d of %d - Local row index starts at %ld with %ld * %d size\n",id,p, local_0_start,local_n0,N1);
  /* create plan for forward DFT */
  plan = fftw_mpi_plan_dft_2d(N0, N1, data, data, MPI_COMM_WORLD,
                  FFTW_FORWARD, FFTW_ESTIMATE);

  /* initialize data to some function my_function(x,y) */
  count = 0;
  for (int i = local_0_start; i < local_0_start + local_n0; ++i) for (int j = 0; j < N1; ++j){
    data[i*N1 + j][0]=local_0_start;
    data[i*N1 + j][1]=i;
    //printf("Processor %d of %d - (%d,%d) - %d - %f + %f i \n",id,p,i+1,j+1,count,data[i*N1 +j][0],data[i*N1 +j][1]);
    count +=1;
  }

  fftw_execute(plan);
  fftw_destroy_plan(plan);

  fftw_free(data);
  MPI_Finalize();
  printf("finalize\n");
  return 0;
}

Я компилирую и запускаю его как

mpicc -I /usr/local/include -L /usr/local/lib  ${PRJ}.c -o ${PRJ} -lfftw3_mpi -lfftw3 -lm 
mpirun -np 2 ./simple_mpi_example

Когда я пытаюсь инициализировать некоторые данные в мои данныеMatrix, приводит меня к ошибке сегментации во время mpirun на MacOS с кодом ошибки

Processor 1 of 2 - Local row index starts at 2 with 2 * 4 size
Processor 0 of 2 - Local row index starts at 0 with 2 * 4 size
[Sanaths-MacBook-Air:82223] *** Process received signal ***
[Sanaths-MacBook-Air:82223] Signal: Segmentation fault: 11 (11)
[Sanaths-MacBook-Air:82223] Signal code:  (0)
[Sanaths-MacBook-Air:82223] Failing at address: 0x0
[Sanaths-MacBook-Air:82223] [ 0] 0   libsystem_platform.dylib            0x00007fff7d48bb3d _sigtramp + 29
[Sanaths-MacBook-Air:82223] *** End of error message ***
--------------------------------------------------------------------------
Primary job  terminated normally, but 1 process returned
a non-zero exit code. Per user-direction, the job has been aborted.
--------------------------------------------------------------------------
--------------------------------------------------------------------------
mpirun noticed that process rank 1 with PID 0 on node Sanaths-MacBook-Air exited on signal 11 (Segmentation fault: 11).
--------------------------------------------------------------------------
make: *** [run] Error 139

Я попытался запустить тот же код на машине с Linux, и он выдал ошибку также, как показано ниже.

Processor 1 of 2 - Local row index starts at 2 with 2 * 4 size
Processor 0 of 2 - Local row index starts at 0 with 2 * 4 size
double free or corruption (!prev)
[sanath-X550LC:05802] *** Process received signal ***
[sanath-X550LC:05802] Signal: Aborted (6)
[sanath-X550LC:05802] Signal code:  (-6)
^C^CMakefile:7: recipe for target 'run' failed

Эта ошибка возникает, только когда я пытаюсь получить доступ к данным, используя вложенные циклы for.Когда я этого не делаю, он работает без ошибок.Это привело меня к мысли, что я могу где-то переступить границы, но это не так.

Любые предложения / советы будут высоко оценены.Спасибо!

1 Ответ

0 голосов
/ 02 февраля 2019

Я полагаю, ваш код основан на этом примере кода MPI ?В примере показано, что каждый локальный процесс обрабатывает срез всего массива:

for (int     i = 0; i < local_n0; ++i) {
    for (int j = 0; j < N1;       ++j) {
        data[i*N1 + j][0] = complex_real_part(local_n0_start + i, j);
        data[i*N1 + j][1] = complex_imag_part(local_n0_start + i, j);
    }
}

Но ваш код предполагает, что он может обращаться к массиву целом , хотя вы тольковыделенная память для локального слайса :

data = fftw_alloc_complex(alloc_local);
memset(data, 0, alloc_local*sizeof(fftw_complex));
...
for (int     i = local_n0; i < local_n0 + local_n0; ++i) {
    for (int j = 0;        j < N1;                  ++j) {
        data[i*N1 + j][0] = complex_real_part(i, j);
        data[i*N1 + j][1] = complex_imag_part(i, j);
    }
}
...