forall_1.f90

来自「linux下编程用 编译软件」· F90 代码 · 共 62 行

! Program to test FORALL construct
program forall_1

   call actual_variable ()
   call negative_stride ()
   call forall_index ()

contains
   subroutine actual_variable ()
      integer:: x = -1
      integer a(3,4)
      j = 100
      
      ! Actual variable 'x' and 'j' used as FORALL index
      forall (x = 1:3, j = 1:4)
         a (x,j) = j
      end forall
      if (any (a.ne.reshape ((/1,1,1,2,2,2,3,3,3,4,4,4/), (/3,4/)))) call abort
      if ((x.ne.-1).or.(j.ne.100)) call abort

      call actual_variable_2 (x, j, a)
   end subroutine

   subroutine actual_variable_2(x, j, a)
      integer x,j,x1,j1
      integer a(3,4), b(3,4)

      ! Actual variable 'x' and 'j' used as FORALL index.
      forall (x=3:1:-1, j=4:1:-1)
         a(x,j) = j
         b(x,j) = j
      end forall

      if (any (a.ne.reshape ((/1,1,1,2,2,2,3,3,3,4,4,4/), (/3,4/)))) call abort
      if (any (b.ne.reshape ((/1,1,1,2,2,2,3,3,3,4,4,4/), (/3,4/)))) call abort
      if ((x.ne.-1).or.(j.ne.100)) call abort
   end subroutine

   subroutine negative_stride ()     
      integer a(3,4)
      integer x, j

      ! FORALL with negative stride
      forall (x = 3:1:-1, j = 4:1:-1)
         a(x,j) = j + x
      end forall
      if (any (a.ne.reshape ((/2,3,4,3,4,5,4,5,6,5,6,7/), (/3,4/)))) call abort
   end subroutine

   subroutine forall_index
      integer a(32,32)

      ! FORALL with arbitrary number indexes
      forall (i1=1:2,i2=1:2,i3=1:2,i4=1:2,i5=1:2,i6=1:2,i7=1:2,i8=1:2,i9=1:2,&
              i10=1:2)
         a(i1+2*i3+4*i5+8*i7+16*i9-30,i2+2*i4+4*i6+8*i8+16*i10-30) = 1
      end forall
      if ((a(5,5).ne.1).or. (a(32,32).ne.1)) call abort
   end subroutine

end