Patch 2/3 of the transition step 1 described in https://discourse.llvm.org/t/rfc-enabling-the-hlfir-lowering-by-default/72778/7. All the modified tests are still here since coverage for the direct lowering to FIR was still needed while it was default. Some already have an HLFIR version, some have not and will need to be ported in step 2 described in the RFC. Note that another 147 lit tests use -emit-fir/-emit-llvm outputs but do not need a flag since the HLFIR/no HLFIR output is the same for what is being tested.
58 lines
2.0 KiB
Fortran
58 lines
2.0 KiB
Fortran
! RUN: bbc -o - -emit-fir -hlfir=false %s | FileCheck %s
|
|
|
|
! Test lowering of elemental calls with array arguments that use array
|
|
! elements as indices.
|
|
! As reported in issue #62981, wrong code was being generated in this case.
|
|
|
|
module test_ops
|
|
implicit none
|
|
interface
|
|
integer elemental function elem_func_i(i)
|
|
integer, intent(in) :: i
|
|
end function
|
|
real elemental function elem_func_r(r)
|
|
real, intent(in) :: r
|
|
end function
|
|
end interface
|
|
|
|
integer :: a(3), b(3), v(3), i, j, k, l
|
|
real :: x(2), y(2), u
|
|
|
|
contains
|
|
! CHECK-LABEL: func @_QMtest_opsPcheck_array_elems_as_indices() {
|
|
subroutine check_array_elems_as_indices()
|
|
! CHECK: %[[A_ADDR:.*]] = fir.address_of(@_QMtest_opsEa) : !fir.ref<!fir.array<3xi32>>
|
|
! CHECK: %[[V_ADDR:.*]] = fir.address_of(@_QMtest_opsEv) : !fir.ref<!fir.array<3xi32>>
|
|
! CHECK: %[[V:.*]] = fir.array_load %[[V_ADDR]](%{{.*}}) : (!fir.ref<!fir.array<3xi32>>, !fir.shape<1>) -> !fir.array<3xi32>
|
|
! CHECK: %[[A:.*]] = fir.array_load %[[A_ADDR]](%{{.*}}) : (!fir.ref<!fir.array<3xi32>>, !fir.shape<1>) -> !fir.array<3xi32>
|
|
! CHECK: fir.do_loop
|
|
forall (i=1:3)
|
|
! CHECK: %{{.*}} = fir.array_fetch %[[V]], %{{.*}} : (!fir.array<3xi32>, index) -> i32
|
|
! CHECK: fir.do_loop
|
|
! CHECK: %[[ELEM:.*]] = fir.array_access %[[A]], %{{.*}} : (!fir.array<3xi32>, index) -> !fir.ref<i32>
|
|
! CHECK: %{{.*}} = fir.call @_QPelem_func_i(%[[ELEM]]){{.*}} : (!fir.ref<i32>) -> i32
|
|
b(i:i) = elem_func_i(a(v(i):v(i)))
|
|
end forall
|
|
end subroutine
|
|
|
|
! CHECK-LABEL: func @_QMtest_opsPcheck_not_assert() {
|
|
subroutine check_not_assert()
|
|
! Implicit path.
|
|
b = 10 + elem_func_i(a)
|
|
|
|
! Expression as argument, instead of variable.
|
|
forall (i=1:3)
|
|
b(i:i) = elem_func_i(a(i:i) + a(i:i))
|
|
end forall
|
|
|
|
! Nested elemental function calls.
|
|
y = elem_func_r(cos(x))
|
|
y = elem_func_r(cos(x) + u)
|
|
|
|
! Array constructors as elemental function arguments.
|
|
y = atan2( (/ (real(i, 4), i = 1, 2) /), &
|
|
real( (/ (i, i = j, k, l) /), 4) )
|
|
end subroutine
|
|
|
|
end module
|