blob: 370532bafaa13cf802b0a2e14450a7fec60eafb5 [file] [log] [blame]
! RUN: %python %S/test_folding.py %s %flang_fc1
! Tests folding of NORM2(), F'2023 16.9.153
module m
! Examples from the standard
logical, parameter :: test_ex1 = norm2([3.,4.]) == 5.
real, parameter :: ex2(2,2) = reshape([1.,3.,2.,4.],[2,2])
real, parameter :: ex2_norm2_1(2) = norm2(ex2, dim=1)
real, parameter :: ex2_1(2) = [3.162277698516845703125,4.472136020660400390625]
logical, parameter :: test_ex2_1 = all(ex2_norm2_1 == ex2_1)
real, parameter :: ex2_norm2_2(2) = norm2(ex2, dim=2)
real, parameter :: ex2_2(2) = [2.2360680103302001953125,5.]
logical, parameter :: test_ex2_2 = all(ex2_norm2_2 == ex2_2)
! 0 3 6 9
! 1 4 7 10
! 2 5 8 11
integer, parameter :: dp = kind(0.d0)
real(dp), parameter :: a(3,4) = &
reshape([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11], shape(a))
real(dp), parameter :: nAll = norm2(a)
real(dp), parameter :: check_nAll = 11._dp * sqrt(sum((a/11._dp)**2))
logical, parameter :: test_all = nAll == check_nAll
real(dp), parameter :: norms1(4) = norm2(a, dim=1)
real(dp), parameter :: check_norms1(4) = [ &
2.236067977499789805051477742381393909454345703125_8, &
7.07106781186547550532850436866283416748046875_8, &
1.2206555615733702069292121450416743755340576171875e1_8, &
1.7378147196982769884243680280633270740509033203125e1_8 ]
logical, parameter :: test_norms1 = all(norms1 == check_norms1)
real(dp), parameter :: norms2(3) = norm2(a, dim=2)
real(dp), parameter :: check_norms2(3) = [ &
1.1224972160321822656214862945489585399627685546875e1_8, &
1.28840987267251261272349438513629138469696044921875e1_8, &
1.4628738838327791427218471653759479522705078125e1_8 ]
logical, parameter :: test_norms2 = all(norms2 == check_norms2)
logical, parameter :: test_normZ = norm2([0.,0.,0.]) == 0.
end