summaryrefslogtreecommitdiff
path: root/gcc/testsuite/gfortran.dg/sum_init_expr.f03
blob: f0cfe958b51658adc6a7239717e4d846ccf4a761 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
! { dg-do run }
! { dg-options "-fno-inline" }
!
! SUM as initialization expression.
!
! This test compares results of simplifier of SUM 
! with the corresponding inlined or library routine(s).
!

  IMPLICIT NONE

  INTEGER, PARAMETER :: imatrix(2,4) = RESHAPE ([ 1, 2, 3, 4, 5, 6, 7, 8 ], [2, 4] )
  INTEGER, PARAMETER :: imatrix_sum = SUM (imatrix)
  INTEGER, PARAMETER :: imatrix_sum_d1(4) = SUM (imatrix, dim=1)
  INTEGER, PARAMETER :: imatrix_sum_d2(2) = SUM (imatrix, dim=2)
  LOGICAL, PARAMETER :: i_equal_sum = ALL ([SUM( imatrix_sum_d1 ) ==  SUM ( imatrix_sum_d2 ), &
                                            SUM( imatrix_sum_d1 ) == imatrix_sum])
  LOGICAL, PARAMETER :: i_empty_sum = SUM(imatrix, mask=.FALSE.) == 0

  REAL, PARAMETER :: rmatrix(2,4) = RESHAPE ([ 1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8 ], [2, 4] )
  REAL, PARAMETER :: rmatrix_sum = SUM (rmatrix)
  REAL, PARAMETER :: rmatrix_sum_d1(4) = SUM (rmatrix, dim=1)
  REAL, PARAMETER :: rmatrix_sum_d2(2) = SUM (rmatrix, dim=2)
  LOGICAL, PARAMETER :: r_equal_sum = ALL ([SUM( rmatrix_sum_d1 ) ==  SUM ( rmatrix_sum_d2 ), &
                                            SUM( rmatrix_sum_d1 ) == rmatrix_sum])
  LOGICAL, PARAMETER :: r_empty_sum = SUM(rmatrix, mask=.FALSE.) == 0.0

  IF (.NOT. ALL ([i_equal_sum, i_empty_sum])) CALL abort()
  IF (.NOT. ALL ([r_equal_sum, r_empty_sum])) CALL abort()

  CALL ilib (imatrix, imatrix_sum)
  CALL ilib_with_dim (imatrix, 1, imatrix_sum_d1)
  CALL ilib_with_dim (imatrix, 2, imatrix_sum_d2)
  CALL rlib (rmatrix, rmatrix_sum)
  CALL rlib_with_dim (rmatrix, 1, rmatrix_sum_d1)
  CALL rlib_with_dim (rmatrix, 2, rmatrix_sum_d2)

CONTAINS
  SUBROUTINE ilib (array, result)
    INTEGER, DIMENSION(:,:), INTENT(in) :: array
    INTEGER, INTENT(in) :: result
    IF (SUM(array) /= result) CALL abort()
  END SUBROUTINE

  SUBROUTINE ilib_with_dim (array, dim, result)
    INTEGER, DIMENSION(:,:), INTENT(in) :: array
    INTEGER, INTENT(iN)                 :: dim
    INTEGER, DIMENSION(:), INTENT(in)   :: result
    IF (ANY (SUM (array, dim=dim) /= result)) CALL abort()
  END SUBROUTINE

  SUBROUTINE rlib (array, result)
    REAL, DIMENSION(:,:), INTENT(in) :: array
    REAL, INTENT(in) :: result
    IF (ABS(SUM(array) - result) > 4e-6) CALL abort()
  END SUBROUTINE

  SUBROUTINE rlib_with_dim (array, dim, result)
    REAL, DIMENSION(:,:), INTENT(in) :: array
    INTEGER, INTENT(iN)              :: dim
    REAL, DIMENSION(:), INTENT(in)   :: result
    IF (ANY (ABS(SUM (array, dim=dim) - result) > 4e-6)) CALL abort()
  END SUBROUTINE
END