存储多维数组fortran 90

时间:2013-11-05 04:09:57

标签: fortran nested-loops fortran90

我有

module mymod
contains
  subroutine mysub(matrix_dum, i_size, j_size)
    integer :: i, j, i_size, j_size
    real(8), dimension(:,:) matrix_dum
    do j=1, j_size
      do i = 1, i_size
        matrix_dum(i,j) = 11.d0*matrix_dum(i,j)
      end do
    end do
  end subroutine mysub
end module mymod 

program main
  use mymod
  implicit none
  real(8), dimension(:,:,:), pointer :: matrix 
  integer :: i, num_matrices, i_size, j_size
  num_matrices = 11
  i_size = 5000
  j_size = 6000
  !only one of them are uncommented in actual practice
  !I have two choices this choice takes a very very long time
  allocate(matrix(num_matrices,i_size,j_size))
  matrix = 11.d0
  do i = 1, num_matrices
    call mysub(matrix(i,:,:),i_size,j_size)
  end do

  !this one does the same job instantly
  allocate(matrix(i_size,j_size,num_matrices))
  matrix = 11.d0
  do i = 1, num_matrices
    call mysub(matrix(:,:,i),i_size,j_size)
  end do
  deallocate(matrix)
end program main`

为什么第二部分按预期工作,但第一部分没有。我知道它必须是Fortran将多维数组存储在一行中并且内存位置依次存储的方式。但我不确定我的语法。我正确地传递了矩阵的地址吗?或者我告诉它复制整个矩阵而不是给它地址。如何将指针传递给多维数组的一部分?这是怎么回事?

1 个答案:

答案 0 :(得分:3)

通常,Fortran处理器按照语言定义的数组元素顺序存储数组 - 这样当您从一个元素移动到另一个元素时,最左边的下标变化最快。

所以矩阵(:,:,1)的所有元素都是连续的 - 彼此相邻。

如果SIZE(矩阵,1)/ = 1,矩阵(1,:,:)的元素将不连续。

对于不是数组元素顺序或不连续的数组的操作通常较慢(例如,它们可能对缓存不太友好,或者在某些情况下可能需要构建中间临时数组)。