42 lines · plain
1! RUN: %python %S/test_folding.py %s %flang_fc12! Tests folding of MATMUL()3module m4 integer, parameter :: ia(2,3) = reshape([1, 2, 2, 3, 3, 4], shape(ia))5 integer, parameter :: ib(3,2) = reshape([1, 2, 3, 2, 3, 4], shape(ib))6 integer, parameter :: ix(*) = [1, 2]7 integer, parameter :: iy(*) = [1, 2, 3]8 integer, parameter :: iab(*,*) = matmul(ia, ib)9 integer, parameter :: ixa(*) = matmul(ix, ia)10 integer, parameter :: iay(*) = matmul(ia, iy)11 logical, parameter :: test_iab = all([iab] == [14, 20, 20, 29])12 logical, parameter :: test_ixa = all(ixa == [5, 8, 11])13 logical, parameter :: test_iay = all(iay == [14, 20])14 15 real, parameter :: ra(*,*) = ia16 real, parameter :: rb(*,*) = ib17 real, parameter :: rx(*) = ix18 real, parameter :: ry(*) = iy19 real, parameter :: rab(*,*) = matmul(ra, rb)20 real, parameter :: rxa(*) = matmul(rx, ra)21 real, parameter :: ray(*) = matmul(ra, ry)22 logical, parameter :: test_rab = all(rab == iab)23 logical, parameter :: test_rxa = all(rxa == ixa)24 logical, parameter :: test_ray = all(ray == iay)25 26 complex, parameter :: za(*,*) = cmplx(ra, -1.)27 complex, parameter :: zb(*,*) = cmplx(rb, -1.)28 complex, parameter :: zx(*) = cmplx(rx, -1.)29 complex, parameter :: zy(*) = cmplx(ry, -1.)30 complex, parameter :: zab(*,*) = matmul(za, zb)31 complex, parameter :: zxa(*) = matmul(zx, za)32 complex, parameter :: zay(*) = matmul(za, zy)33 logical, parameter :: test_zab = all([zab] == [(11,-12),(17,-15),(17,-15),(26,-18)])34 logical, parameter :: test_zxa = all(zxa == [(3,-6),(6,-8),(9,-10)])35 logical, parameter :: test_zay = all(zay == [(11,-12),(17,-15)])36 37 logical, parameter :: la(16, 4) = reshape([((iand(shiftr(j,k),1)/=0, j=0,15), k=0,3)], shape(la))38 logical, parameter :: lb(4, 16) = transpose(la)39 logical, parameter :: lab(16, 16) = matmul(la, lb)40 logical, parameter :: test_lab = all([lab] .eqv. [((iand(k,j)/=0, k=0,15), j=0,15)])41end42