1! RUN: %python %S/test_folding.py %s %flang_fc1 2! Tests folding of MATMUL() 3module m 4 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(*,*) = ia 16 real, parameter :: rb(*,*) = ib 17 real, parameter :: rx(*) = ix 18 real, parameter :: ry(*) = iy 19 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)]) 41end 42