我有
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将多维数组存储在一行中并且内存位置依次存储的方式。但我不确定我的语法。我正确地传递了矩阵的地址吗?或者我告诉它复制整个矩阵而不是给它地址。如何将指针传递给多维数组的一部分?这是怎么回事?
答案 0 :(得分:3)
通常,Fortran处理器按照语言定义的数组元素顺序存储数组 - 这样当您从一个元素移动到另一个元素时,最左边的下标变化最快。
所以矩阵(:,:,1)的所有元素都是连续的 - 彼此相邻。
如果SIZE(矩阵,1)/ = 1,矩阵(1,:,:)的元素将不连续。
对于不是数组元素顺序或不连续的数组的操作通常较慢(例如,它们可能对缓存不太友好,或者在某些情况下可能需要构建中间临时数组)。