使用具有4个线程的MPI_SENDRECV的问题

时间:2019-04-01 02:36:10

标签: parallel-processing fortran mpi

作为一个最小的问题,我正在尝试在4个处理器之间发送一个整数:0-> 3(等级0从等级3发送和接收),2-> 1,1 -> 2、3->0。它永远不会完成执行并挂起,可能正在等待其他线程的响应。

我正在使用mpif90 ...编译代码并使用mpiexec -np 4 ...运行。以下是最小片段:

program sendrecv
  implicit none
  include "mpif.h"
  integer :: foo, bar
  integer :: mpi_rank, mpi_size, ierr
  integer :: mpi_sendto, mpi_recvfrom
  integer :: istat(MPI_STATUS_SIZE), status, i

  call MPI_INIT(ierr)
  call MPI_COMM_SIZE(MPI_COMM_WORLD, mpi_size, ierr)
  call MPI_COMM_RANK(MPI_COMM_WORLD, mpi_rank, ierr)
  print *, "SENDING..."

  if (mpi_rank .eq. 0) then
    mpi_sendto = 3; mpi_recvfrom = 3
  else if (mpi_rank .eq. 1) then
    mpi_sendto = 2; mpi_recvfrom = 2
  else if (mpi_rank .eq. 2) then
    mpi_sendto = 1; mpi_recvfrom = 1
  else
    mpi_sendto = 0; mpi_recvfrom = 0
  end if

  foo = mpi_rank
  do i = 1, 5
    foo = mpi_rank
    call MPI_SENDRECV(foo, 1,&
                    & MPI_INTEGER, mpi_sendto, mpi_rank * 10 + i,&
                    & bar, 1,&
                    & MPI_INTEGER, mpi_recvfrom, mpi_rank * 10 + i,&
                    & MPI_COMM_WORLD, istat, ierr)
  end do

  print *, "...DONE"
  call MPI_FINALIZE(ierr)
end

我不太了解为什么该程序挂起,也许我丢失了某些东西或做了一些真正错误的事情。如果我正确理解,MPI_SENDRECV就是非阻塞sendrecv的两个wait-s。那样的话,例如,如果rank=0发送到rank=3,从它那里接收应该没有任何问题,对吧?

我尝试从不同的线程发送/接收,即:

  if (mpi_rank .eq. 0) then
    mpi_sendto = 1; mpi_recvfrom = 3
  else if (mpi_rank .eq. 1) then
    mpi_sendto = 2; mpi_recvfrom = 0
  else if (mpi_rank .eq. 2) then
    mpi_sendto = 3; mpi_recvfrom = 1
  else
    mpi_sendto = 0; mpi_recvfrom = 2
  end if

仍然无法正常工作。

UPD 如前所述,标记在执行SENDRECV时应该是相同的,但是如果在循环中进行此调用,类似的标记将无济于事(请参见修改后的代码)。旧版本:

call MPI_SENDRECV(foo, 1,&
                    & MPI_INTEGER, mpi_sendto, 200,&
                    & bar, 1,&
                    & MPI_INTEGER, mpi_recvfrom, 100,&
                    & MPI_COMM_WORLD, status, ierr)

UPD#2 实际上,如果有人感兴趣,我发现discussion正是关于为什么SENDRECV -s有时会死锁的问题。

1 个答案:

答案 0 :(得分:2)

“线程”一词在这里令人误解,您应该谈论MPI任务或MPI流程(两者是等效的)。

根本原因是标签不匹配。您使用标签200发送,但使用标签100接收。

此外,您应该使用istat代替status作为MPI_Sendrecv()的状态参数。

这是修复程序的方法

call MPI_SENDRECV(foo, 1,&
                & MPI_INTEGER, mpi_sendto, 200,&
                & bar, 1,&
                & MPI_INTEGER, mpi_recvfrom, 200,&
                & MPI_COMM_WORLD, istat, ierr)