避免在 Fortran DO 循环中使用 IF 语句 [重复]

Question

在我的 Fortran 程序中，我有类似这个循环的东西

do i = 1,imax
    if(period) then
        call g_period()
    else
        call g()
    endif
 enddo

我在 do 循环中有几个地方，有时还有几个 If 循环，用于检查我是否应该调用函数 X 或 Y。所有这些的共同点是在输入文件中定义了句点，并且在整个运行时间。 有没有办法避免 if 语句 - 因为我的速度被所有检查的东西杀死了？虽然，我想维护一个通用的原因，即能够使用 period = true 运行程序一次，而使用 period = false 运行其他时间。

Answer 1

如果速度真的很重要，您总是可以编写两个不同的版本。一真一假：

if(period) then
  do i = 1,imax
    call g_period()
  end do
else
  do i = 1,imax
    call g()
  end do
end if

你甚至可以拥有完全独立的子程序和一切。

当您调用子例程 g() 和替代例程时，它们会做大量工作，可能不会对 if 造成太大的性能损失。但是，如果你做一些事情，比如循环一个包含许多点的大型 3D 数组，并在每个点上做一些数学运算，只有少数例外，那么在任何地方计算它并在之后为那些少数例外点撤消它可能会更快，而不是在紧密循环中引入一个分支。