57 lines
2 KiB
Fortran
57 lines
2 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
|