使用 MPI_Bcast 传递 3d 数组

Question

我正在尝试使用MPI_Bcast 将 3D 数组传递给所有其他进程（在 FORTRAN 77 中）。 v1 是一个普通的块数组。我也不确定是否需要将公共数组v1 的计算值广播给所有其他进程，否则它们会因为通用而在每个进程中更改。以下是相关代码：

  parameter (nprocz=48,nzro=1)

    do i=i101,i102
      dist  = 0.015*float(i-iv0)
      adamp = exp(-dist*dist)
      do j = je0, je1-1
      do k = ke0, ke1
        v1(k,j,i) = v1(k,j,i)*adamp
      end do
      end do
    end do

    nmpi01=floor((iv0-ie0-nzro)/(nprocz-1))
    if (mpirank .le. nprocz-2) then
       i101=ie0+(mpirank*nmpi01)
       i102=ie0+(mpirank+1)*nmpi01-1
    else
       i101=ie0+(mpirank*nmpi01)
       i102=iv0-1
    endif

   MPI_Bcast(v1(:,:,i101:i102),(ke1-ke0+1)*(je1-je0)*(i102-i101+1)
 & ,MPI_FLOAT,mpirank,MPI_COMM_WORLD,ierr01)

我收到错误消息：

PGFTN-S-0081-Matrix/vector v1 illegal as subprogram argument

传入的数组大小是正确的。有什么意见吗？

---

我更正了代码并遍历了排名并计算了每个排名中 rcount 和 displs 的所有元素：

integer :: myscount, myi101

do rank = 0, nprocz-1
  nmpi01=floor((iv0-ie0-nzro)/(nprocz-1))
  if (rank .le. nprocz-2) then
    i101=ie0+(rank*nmpi01)
    i102=ie0+(rank+1)*nmpi01-1
  else
    i101=ie0+(rank*nmpi01)
    i102=iv0-1
  endif
  scount=(i102-i101+1)*(je1-je0)*(ke1-ke0+1)
  rcount(rank+1)=scount
  displs(rank+1)=rank*scount+1
  if (rank .eq. mpirank) then
    myscount = scount
    myi101 = i101
 end if
end do

scount = myscount
i101 = myi101

call mpi_allgatherv(...)

但仍然是错误的结果。 1-在我的情况下，每个部分的结果都用于下一部分，尤其是在mpi_allgatherv之后。所以我需要在每个mpi_allgatherv之后添加mpi_barrier吗？ 2-应该使用 mpi_in_place 吗？考虑我只有一个 3d 数组 v1，每个子数组 v1(1,1,i) 是由某个过程计算的，我想将计算的子数组放在同一数组的适当部分。 3-我想我应该有displs(i) = sum(rcount(1:i-1))+1 for i=>2 考虑到在 fortran77 中总是 displs(1)=1。所以我更正了这个：循环之前displs(1)=1，循环内部displs(rank+2)=rank*scount+1和循环之后displs(nprocz+1)=0。我说的对吗？

Answer 1

我记得，Fortran 77 对数组下标的限制比 Fortran 90 更严格，而 pgftn 是 Fortran 77 编译器。我会尝试将v1(1,1,i101) 传递给mpi_bcast，而不是v1(:,:,i101:i102)。 （或者使用带有“-Mfixed”标志的 pgf95。）

如果每个进程都需要查看 v1，那么您确实需要使用 MPI 进行通信。 MPI 任务之间不共享变量，即使是公共块中的任务也不共享。但是，如果每个进程都在计算 v1 的不同部分，那么每个进程都需要其他进程的一部分，那么您不能使用 mpi_bcast 来执行此操作；请改用mpi_allgather。

另外，如上所述，当您使用 MPI 程序时，您应该 call 它们，因为它们是子程序。