Question

我的问题是，当我在Fortran中使用mpi方案时，我不知道如何调用子例程。我写了这个名为TRY.f90的小代码，其中有一个名为CONCENTRATION.f90的子程序。我应该如何更改CONCENTRATION.f90以使代码有效？

PROGRAM TRY
USE MPI

integer status(mpi_status_size)
INTEGER                 I, J, K, II, IERR, MY_ID, NUM_PROCS, PSP
INTEGER , PARAMETER             :: GRIDX =64, GRIDY=64 
REAL    , DIMENSION(gridx,gridy)    :: PSI
PSI=0
PRINT*, 'VARIABLE'

CALL MPI_INIT(IERR)
CALL MPI_COMM_RANK(MPI_COMM_WORLD,MY_ID,IERR)
CALL MPI_COMM_SIZE(MPI_COMM_WORLD,NUM_PROCS,IERR)

CALL CONCENTRATION(GRIDX, GRIDY, NUM_PROCS, MY_ID , PSI)

IF (MY_ID .NE. 0) THEN
    CALL mpi_send( PSI(1+MY_ID*GRIDX/NUM_PROCS:(MY_ID+1)*GRIDX/NUM_PROCS:1,1:GRIDY:1),&
    (GRIDX/NUM_PROCS)*GRIDY,mpi_real, 0,10,mpi_comm_world,ierr)
END IF
IF (MY_ID .EQ. 0) THEN
    DO II=1,NUM_PROCS-1
        CALL mpi_recv(PSI(1+II*GRIDX/NUM_PROCS:(II+1)*GRIDX/NUM_PROCS:1,1:GRIDY:1),&
        (GRIDX/NUM_PROCS)*GRIDY,mpi_real, &
        II,10,mpi_comm_world,status,ierr)
    END DO
 END IF

 CALL MPI_FINALIZE(IERR)
 END PROGRAM TRY

我正在使用名为CONCENTRATION.f90的子程序：

SUBROUTINE CONCENTRATION(GRIDX, GRIDY, NUM_PROCS, MY_ID , PSI)
implicit none

INTEGER*8, INTENT(IN)               :: GRIDX, GRIDY
INTEGER  , INTENT(IN)               :: NUM_PROCS, MY_ID
REAL*8   , DIMENSION(GRIDX,GRIDY), INTENT(OUT)  :: PSI
INTEGER*8                          I, J

DO I=1+MY_ID*GRIDX/NUM_PROCS, (MY_ID+1)*GRIDX/NUM_PROCS
    DO J=1,GRIDY
        PSI(I,J)=2.0
    END DO
END DO

END SUBROUTINE CONCENTRATION

代码目前给我错误，因为我认为我应该对子程序CONCENTRATION.f90进行一些更改。或者我也应该改变我调用子程序的方式。

你能告诉我这些变化是什么吗？感谢您的帮助

Answer 1

由于类型不匹配，您的程序会出现段错误。在主程序中，您已将PSI声明为REAL：

的数组

REAL    , DIMENSION(gridx,gridy)    :: PSI

在CONCENTRATION子例程中，您使用其他类型的REAL*8：

REAL*8   , DIMENSION(GRIDX,GRIDY), INTENT(OUT)  :: PSI

默认情况下，REAL长度为4个字节，而REAL*8（或DOUBLE PRECISION或REAL(KIND=8)）长度为8个字节。所以你给CONCENTRATION一个比它认为的小2倍的数组，从NUM_PROCS/2开始的所有等级都写在PSI数组的末尾之后，从而导致段错误。如果只运行一个进程，那么即使是0级也会出现段错误。

您还应该阅读有关MPI集体操作的内容。 MPI_GATHER和MPI_GATHERV完全符合您在多个发送和接收时所要实现的目标。

Answer 2

唯一的变化是将concentration声明为reentrant。这可能是Fortran 90的默认设置。（我的大部分经验是F77，reentrant不是默认值。）

如何在MPI并行化的fortran中调用子程序？

2 个答案: