特别是在 Fortran 的 MPI 中,仅在根进程上定义数组是否可能并且是一个不错的选择?例如这样的事情:
program test
implicit none
include 'mpif.h'
all mpi_init(ierr)
call mpi_comm_rank(mpi_comm_world,myid,ierr)
call mpi_comm_size(mpi_comm_world,numproc,ierr)
if (myid .eq. 0) then
complex(8), dimension(:,:), allocatable :: array
end if
...
if (myid .eq. 0) then
allocate(array(2,2))
end if
...
end program
你可以猜到,我已经尝试过这个,但它不起作用,因为在 Fortran 中声明需要位于顶部。但我希望有办法解决这个问题?
这样,数组也不会占用我的“虚拟”内存,对吗?或者我误解了什么?
正如您所指出的,您不能在中包含声明语句IF
-块或过程的声明块之后的任何位置。但是,将数组声明为ALLOCATABLE
在所有进程上,并且仅在某些进程上分配是允许的,在我看来,这是一个不错的选择。
! Declare as allocatable on all processes
complex(8), dimension(:,:), allocatable :: array
...
! Allocate only on some
if (myid .eq. 0) then
allocate(array(2,2))
end if
这样,程序就不会消耗其他进程的任何额外内存。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)