summaryrefslogtreecommitdiff
path: root/gcc/testsuite/gfortran.dg/matmul_6.f90
blob: 737c5c437596da36b3313419caf8e4d262dc7648 (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 }
! PR 34566 - logical matmul used to give the wrong result.
! We check this by running through every permutation in
! multiplying two 3*3 matrices, and all permutations of multiplying
! a 3-vector and a 3*3 matrices  and checking against equivalence
! with integer matrix multiply.
program main
  implicit none
  integer, parameter :: ki=4
  integer, parameter :: dimen=3
  integer :: i, j, k
  real, dimension(dimen,dimen) :: r1, r2
  integer, dimension(dimen,dimen) :: m1, m2
  logical(kind=ki), dimension(dimen,dimen) :: l1, l2
  logical(kind=ki), dimension(dimen*dimen) :: laux
  logical(kind=ki), dimension(dimen) :: lv
  integer, dimension(dimen) :: iv

  do i=0,2**(dimen*dimen)-1
     forall (k=1:dimen*dimen)
        laux(k) = btest(i, k-1)
     end forall
     l1 = reshape(laux,shape(l1))
     m1 = ltoi(l1)

     ! Check matrix*matrix multiply
     do j=0,2**(dimen*dimen)-1
        forall (k=1:dimen*dimen)
           laux(k) = btest(i, k-1)
        end forall
        l2 = reshape(laux,shape(l2))
        m2 = ltoi(l2)
        if (any(matmul(l1,l2) .neqv. (matmul(m1,m2) /= 0))) then
          call abort
        end if
     end do

     ! Check vector*matrix and matrix*vector multiply.
     do j=0,2**dimen-1
        forall (k=1:dimen)
           lv(k) = btest(j, k-1)
        end forall
        iv = ltoi(lv)
        if (any(matmul(lv,l1) .neqv. (matmul(iv,m1) /=0))) then
          call abort
        end if
        if (any(matmul(l1,lv) .neqv. (matmul(m1,iv) /= 0))) then
          call abort
        end if
     end do
  end do

contains
  elemental function ltoi(v)
    implicit none
    integer :: ltoi
    real :: rtoi
    logical(kind=4), intent(in) :: v
    if (v) then
       ltoi = 1
    else
       ltoi = 0
    end if
  end function ltoi

end program main