Fortran中大型数组传子程序的高效方式及内存优化咨询

我有一个用Fortran编写的物理程序，用于处理用户输入尺寸的大型3D域（维度为A×B×C）计算，涉及多个可分配大型变量数组。当前使用Intel编译器编译，同时需要适配gfortran的高效方案，程序运行于数千核的并行计算集群。

程序核心示例代码如下：

program hello
implicit none

real, allocatable, dimension(:,:,:)    ::    big_array
integer      ::      A, B, C

! 根据用户输入完成计算并定义相关参数

allocate(big_array(1:A,1:B,1:C)) 

call do_something_to_big_array(big_array)

end program hello

问题1：大型数组的高效传递方式 #

当前向子程序do_something_to_big_array传递数组的方式如下：

subroutine do_something_to_big_array(big_array)
implicit none
real, dimension(1:A, 1:B, 1:C)     ::    big_array

! 修改big_array的计算逻辑

end subroutine

由于该子程序会被迭代调用数千次，有没有更高效的最佳实践方案？

问题2：局部大数组的内存优化 #

部分子程序使用仅在局部作用域内的全域尺寸数组，定义为局部变量会显著增加内存占用，示例代码如下：

subroutine do_something_else(big_array)
implicit none

real, dimension(1:A,1:B,1:C)    ::    big_array
real, dimension(1:A,1:B,1:C)    ::    temp_array ! 仅在本子程序中使用的临时数组

end subroutine

这类子程序同样被频繁调用，将数组定义为全局作用域是否更高效？

解决方案 #

针对问题1：数组传递的高效优化

• 改用假定形状数组：当前子程序依赖全局变量A/B/C指定数组维度，耦合性高且编译器优化受限。将参数声明改为real, intent(inout), dimension(:,:,:) :: big_array，编译器可直接利用数组的描述符（包含维度、内存 stride 等信息），完全避免数组复制，同时适配任意尺寸的数组，灵活性更强。
• 明确参数意图：添加intent(inout)声明，告诉编译器数组是输入输出型参数，便于编译器做精准优化（比如消除不必要的内存拷贝）。
• 并行环境适配：Fortran默认采用列优先内存布局，假定形状数组会保留这一连续内存信息，对OpenMP/MPI等并行框架的优化更友好。Intel和gfortran对假定形状数组的优化都很成熟，不会产生性能损耗。

针对问题2：局部大数组的内存优化

• 不推荐全局变量：全局变量会破坏代码封装性，并行环境下还可能引发数据竞争（即使做线程私有处理，管理成本也很高），且全局数组会一直驻留内存，并不会真正减少内存占用。
• 推荐方案：模块缓存可分配数组
  将临时数组放在独立模块中，第一次调用时分配内存，后续调用直接复用，程序退出时释放。示例代码如下：

  module temp_array_cache
    implicit none
    real, allocatable, dimension(:,:,:) :: temp_array
  contains
  end module temp_array_cache
  
  subroutine do_something_else(big_array)
    use temp_array_cache
    implicit none
    real, dimension(:,:,:), intent(inout) :: big_array
    integer :: dim1, dim2, dim3
  
    dim1 = size(big_array, 1)
    dim2 = size(big_array, 2)
    dim3 = size(big_array, 3)
  
    ! 仅在未分配时初始化数组
    if (.not. allocated(temp_array)) allocate(temp_array(dim1, dim2, dim3))
    
    ! 此处使用temp_array完成计算
  end subroutine do_something_else
  

  这种方式既避免了频繁分配/释放内存的开销，又保持了代码的封装性。并行环境下，可通过!$omp threadprivate(temp_array)（需编译器支持）将数组设为线程私有，避免竞争。
• 备选方案：调用方传递临时数组
  如果上层调用者可以复用同一个临时数组，将temp_array作为参数传递给子程序，把内存控制权交给调用方，灵活性更高，也能避免重复分配。

内容的提问来源于stack exchange，提问作者mgmf46