fortran - 为什么 Fortran 内部函数 "spread"通常比显式迭代慢

Question

我使用地球物理模型，常见的情况是需要将 2D 数据与 3D 数据相乘、相加等。下面是一个例子。

module benchmarks
  implicit none
  integer, parameter :: n=500
  integer :: k
  real :: d2(n,n)
  real :: d3(n,n,n)
  contains
  ! Iteration
  subroutine benchmark_a(res)
    real, intent(out) :: res(n,n,n)
    do k = 1, size(d3,3)
      res(:,:,k) = d2*d3(:,:,k)
    end do
  end subroutine
  ! Spread
  subroutine benchmark_b(res)
    real, intent(out) :: res(n,n,n)
    res = d3*spread(d2, 3, size(d3,3))
  end subroutine
end module

program main
  use benchmarks
  real :: t, tarray(2)
  real :: res(n,n,n)
  call random_number(d2)
  call random_number(d3)
  ! Iteration
  call dtime(tarray, t)
  call benchmark_a(res)
  call dtime(tarray, t)
  write(*,*) 'Iteration', t
  ! Spread
  call dtime(tarray, t)
  call benchmark_b(res)
  call dtime(tarray, t)
  write(*,*) 'Spread', t
end program

当我以不同的维度大小n运行此程序时，我通常发现spread要慢得多；例如:

Spread   2.09942889
Iteration  0.458283991

有谁知道为什么 spread 方法比显式 for 循环(我认为一般情况下要不惜一切代价避免)慢得多？

Answer 1

这里的基本答案是“不是”。也许对于特定的编译器和特定的环境，内在函数不如显式 DO 循环优化得更好，但不一定是这样。我使用 ifort 19 进行了测试，即使在默认优化级别，SPREAD 内在函数和显式循环也会生成类似的代码，当我更正程序以使用结果时，内在函数会更快。

迭代 0.2187500 0.1376885
价差 9.3750000E-02 0.1376885

我还要提醒您(正如我在对您的问题的评论中所做的那样)，简单化的基准程序通常不会衡量作者认为它们所做的事情。最常见的错误(原始示例和修订示例都显示)是，从未使用被测工作的结果，因此足够聪明的编译器可以简单地蒸发整个操作。事实上，当我使用 ifort 19 构建两个测试用例时，编译器完全删除了所有工作，只留下计时代码。不用说，它运行得相当快。

  implicit none
  integer, parameter :: n=500
  integer :: k
  real :: d2(n,n)
  real :: d3(n,n,n)
  contains
  ! Iteration
  subroutine benchmark_a(res)
    real, intent(out) :: res(n,n,n)
    do k = 1, size(d3,3)
      res(:,:,k) = d2*d3(:,:,k)
    end do
  end subroutine
  ! Spread
  subroutine benchmark_b(res)
    real, intent(out) :: res(n,n,n)
    res = d3*spread(d2, 3, size(d3,3))
  end subroutine
end module

program main
  use benchmarks
  real :: tstart,tend
  real :: res(n,n,n)
  call random_number(d2)
  call random_number(d3)
  ! Iteration
  call cpu_time(tstart)
  call benchmark_a(res)
  call cpu_time(tend)
  write(*,*) 'Iteration', tend-tstart, res(10,10,10)
  ! Spread
  call cpu_time(tstart)
  call benchmark_b(res)
  call cpu_time(tend)
  write(*,*) 'Spread', tend-tstart, res(10,10,10)
end program```