58 lines · plain
1! RUN: bbc -o - -emit-fir -hlfir=false %s | FileCheck %s2 3! Test lowering of elemental calls with array arguments that use array4! elements as indices.5! As reported in issue #62981, wrong code was being generated in this case.6 7module test_ops8 implicit none9 interface10 integer elemental function elem_func_i(i)11 integer, intent(in) :: i12 end function13 real elemental function elem_func_r(r)14 real, intent(in) :: r15 end function16 end interface17 18 integer :: a(3), b(3), v(3), i, j, k, l19 real :: x(2), y(2), u20 21contains22! CHECK-LABEL: func @_QMtest_opsPcheck_array_elems_as_indices() {23subroutine check_array_elems_as_indices()24! CHECK: %[[A_ADDR:.*]] = fir.address_of(@_QMtest_opsEa) : !fir.ref<!fir.array<3xi32>>25! CHECK: %[[V_ADDR:.*]] = fir.address_of(@_QMtest_opsEv) : !fir.ref<!fir.array<3xi32>>26! CHECK: %[[V:.*]] = fir.array_load %[[V_ADDR]](%{{.*}}) : (!fir.ref<!fir.array<3xi32>>, !fir.shape<1>) -> !fir.array<3xi32>27! CHECK: %[[A:.*]] = fir.array_load %[[A_ADDR]](%{{.*}}) : (!fir.ref<!fir.array<3xi32>>, !fir.shape<1>) -> !fir.array<3xi32>28! CHECK: fir.do_loop29 forall (i=1:3)30! CHECK: %{{.*}} = fir.array_fetch %[[V]], %{{.*}} : (!fir.array<3xi32>, index) -> i3231! CHECK: fir.do_loop32! CHECK: %[[ELEM:.*]] = fir.array_access %[[A]], %{{.*}} : (!fir.array<3xi32>, index) -> !fir.ref<i32>33! CHECK: %{{.*}} = fir.call @_QPelem_func_i(%[[ELEM]]){{.*}} : (!fir.ref<i32>) -> i3234 b(i:i) = elem_func_i(a(v(i):v(i)))35 end forall36end subroutine37 38! CHECK-LABEL: func @_QMtest_opsPcheck_not_assert() {39subroutine check_not_assert()40 ! Implicit path.41 b = 10 + elem_func_i(a)42 43 ! Expression as argument, instead of variable.44 forall (i=1:3)45 b(i:i) = elem_func_i(a(i:i) + a(i:i))46 end forall47 48 ! Nested elemental function calls.49 y = elem_func_r(cos(x))50 y = elem_func_r(cos(x) + u)51 52 ! Array constructors as elemental function arguments.53 y = atan2( (/ (real(i, 4), i = 1, 2) /), &54 real( (/ (i, i = j, k, l) /), 4) )55end subroutine56 57end module58