42 lines
1.9 KiB
Fortran
42 lines
1.9 KiB
Fortran
|
! RUN: %python %S/test_folding.py %s %flang_fc1
|
||
|
|
! Tests folding of MATMUL()
|
||
|
|
module m
|
||
|
|
integer, parameter :: ia(2,3) = reshape([1, 2, 2, 3, 3, 4], shape(ia))
|
||
|
|
integer, parameter :: ib(3,2) = reshape([1, 2, 3, 2, 3, 4], shape(ib))
|
||
|
|
integer, parameter :: ix(*) = [1, 2]
|
||
|
|
integer, parameter :: iy(*) = [1, 2, 3]
|
||
|
|
integer, parameter :: iab(*,*) = matmul(ia, ib)
|
||
|
|
integer, parameter :: ixa(*) = matmul(ix, ia)
|
||
|
|
integer, parameter :: iay(*) = matmul(ia, iy)
|
||
|
|
logical, parameter :: test_iab = all([iab] == [14, 20, 20, 29])
|
||
|
|
logical, parameter :: test_ixa = all(ixa == [5, 8, 11])
|
||
|
|
logical, parameter :: test_iay = all(iay == [14, 20])
|
||
|
|
|
||
|
|
real, parameter :: ra(*,*) = ia
|
||
|
|
real, parameter :: rb(*,*) = ib
|
||
|
|
real, parameter :: rx(*) = ix
|
||
|
|
real, parameter :: ry(*) = iy
|
||
|
|
real, parameter :: rab(*,*) = matmul(ra, rb)
|
||
|
|
real, parameter :: rxa(*) = matmul(rx, ra)
|
||
|
|
real, parameter :: ray(*) = matmul(ra, ry)
|
||
|
|
logical, parameter :: test_rab = all(rab == iab)
|
||
|
|
logical, parameter :: test_rxa = all(rxa == ixa)
|
||
|
|
logical, parameter :: test_ray = all(ray == iay)
|
||
|
|
|
||
|
|
complex, parameter :: za(*,*) = cmplx(ra, -1.)
|
||
|
|
complex, parameter :: zb(*,*) = cmplx(rb, -1.)
|
||
|
|
complex, parameter :: zx(*) = cmplx(rx, -1.)
|
||
|
|
complex, parameter :: zy(*) = cmplx(ry, -1.)
|
||
|
|
complex, parameter :: zab(*,*) = matmul(za, zb)
|
||
|
|
complex, parameter :: zxa(*) = matmul(zx, za)
|
||
|
|
complex, parameter :: zay(*) = matmul(za, zy)
|
||
|
|
logical, parameter :: test_zab = all([zab] == [(11,-12),(17,-15),(17,-15),(26,-18)])
|
||
|
|
logical, parameter :: test_zxa = all(zxa == [(3,-6),(6,-8),(9,-10)])
|
||
|
|
logical, parameter :: test_zay = all(zay == [(11,-12),(17,-15)])
|
||
|
|
|
||
|
|
logical, parameter :: la(16, 4) = reshape([((iand(shiftr(j,k),1)/=0, j=0,15), k=0,3)], shape(la))
|
||
|
|
logical, parameter :: lb(4, 16) = transpose(la)
|
||
|
|
logical, parameter :: lab(16, 16) = matmul(la, lb)
|
||
|
|
logical, parameter :: test_lab = all([lab] .eqv. [((iand(k,j)/=0, k=0,15), j=0,15)])
|
||
|
|
end
|