flang/test/Lower/HLFIR/parent-component-ref.f90

   1 ! Test lowering of parent component references to HLFIR.
   2 ! RUN: bbc -emit-fir -hlfir -polymorphic-type -o - %s -I nw | FileCheck %s
   3
   4 module pc_types
   5   type t
   6     integer :: i
   7   end type
   8   type, extends(t) :: t2
   9     integer :: j
  10   end type
  11 interface
  12 subroutine takes_t_type_array(x)
  13   import :: t
  14   type(t) :: x(:)
  15 end subroutine
  16 subroutine takes_t_class_array(x)
  17   import :: t
  18   class(t) :: x(:)
  19 end subroutine
  20 subroutine takes_int_array(x)
  21   integer :: x(:)
  22 end subroutine
  23 end interface
  24 end module
  25
  26 subroutine test_ignored_inner_parent_comp(x)
  27  use pc_types
  28  type(t2) :: x
  29  call takes_int(x%t%i)
  30 end subroutine
  31 ! CHECK-LABEL: func.func @_QPtest_ignored_inner_parent_comp(
  32 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}Ex"
  33 ! CHECK:  %[[VAL_2:.*]] = hlfir.designate %[[VAL_1]]#0{"i"}   : (!fir.ref<!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>) -> !fir.ref<i32>
  34 ! CHECK:  fir.call @_QPtakes_int(%[[VAL_2]])
  35
  36 subroutine test_ignored_inner_parent_comp_2(x)
  37  use pc_types
  38  type(t2) :: x(:)
  39  call takes_int_array(x%t%i)
  40 end subroutine
  41 ! CHECK-LABEL: func.func @_QPtest_ignored_inner_parent_comp_2(
  42 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}Ex"
  43 ! CHECK:  %[[VAL_2:.*]] = arith.constant 0 : index
  44 ! CHECK:  %[[VAL_3:.*]]:3 = fir.box_dims %[[VAL_1]]#0, %[[VAL_2]] : (!fir.box<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>, index) -> (index, index, index)
  45 ! CHECK:  %[[VAL_4:.*]] = fir.shape %[[VAL_3]]#1 : (index) -> !fir.shape<1>
  46 ! CHECK:  %[[VAL_5:.*]] = hlfir.designate %[[VAL_1]]#0{"i"}   shape %[[VAL_4]] : (!fir.box<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>, !fir.shape<1>) -> !fir.box<!fir.array<?xi32>>
  47 ! CHECK:  fir.call @_QPtakes_int_array(%[[VAL_5]])
  48
  49 subroutine test_leaf_parent_ref(x)
  50  use pc_types
  51  type(t2) :: x
  52  call takes_parent(x%t)
  53 end subroutine
  54 ! CHECK-LABEL: func.func @_QPtest_leaf_parent_ref(
  55 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}Ex"
  56 ! CHECK:  %[[VAL_2:.*]] = hlfir.parent_comp %[[VAL_1]]#0 : (!fir.ref<!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>) -> !fir.ref<!fir.type<_QMpc_typesTt{i:i32}>>
  57 ! CHECK:  fir.call @_QPtakes_parent(%[[VAL_2]])
  58
  59 subroutine test_leaf_parent_ref_array(x)
  60  use pc_types
  61  class(t2) :: x(42:)
  62 ! CHECK-LABEL: func.func @_QPtest_leaf_parent_ref_array(
  63 ! CHECK:  %[[VAL_4:.*]]:2 = hlfir.declare {{.*}}Ex"
  64  call takes_t_type_array(x%t)
  65 ! CHECK:  %[[VAL_5:.*]] = arith.constant 0 : index
  66 ! CHECK:  %[[VAL_6:.*]]:3 = fir.box_dims %[[VAL_4]]#0, %[[VAL_5]] : (!fir.class<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>, index) -> (index, index, index)
  67 ! CHECK:  %[[VAL_7:.*]] = fir.shape %[[VAL_6]]#1 : (index) -> !fir.shape<1>
  68 ! CHECK:  %[[VAL_8:.*]] = hlfir.parent_comp %[[VAL_4]]#0 shape %[[VAL_7]] : (!fir.class<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>, !fir.shape<1>) -> !fir.box<!fir.array<?x!fir.type<_QMpc_typesTt{i:i32}>>>
  69 ! CHECK:  fir.call @_QPtakes_t_type_array(%[[VAL_8]])
  70  call takes_t_class_array(x%t)
  71 ! CHECK:  %[[VAL_12:.*]] = hlfir.parent_comp %[[VAL_4]]#0 shape %{{.*}} : (!fir.class<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>, !fir.shape<1>) -> !fir.box<!fir.array<?x!fir.type<_QMpc_typesTt{i:i32}>>>
  72 ! CHECK:  %[[VAL_13:.*]] = fir.convert %[[VAL_12]] : (!fir.box<!fir.array<?x!fir.type<_QMpc_typesTt{i:i32}>>>) -> !fir.class<!fir.array<?x!fir.type<_QMpc_typesTt{i:i32}>>>
  73 ! CHECK:  fir.call @_QPtakes_t_class_array(%[[VAL_13]])
  74 end subroutine
  75
  76 subroutine test_parent_section_leaf_array(x)
  77  use pc_types
  78  class(t2) :: x(:)
  79  call takes_t_type_array(x(2:11)%t)
  80 end subroutine
  81 ! CHECK-LABEL: func.func @_QPtest_parent_section_leaf_array(
  82 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}Ex"
  83 ! CHECK:  %[[VAL_7:.*]] = hlfir.designate %[[VAL_1]]#0 ({{.*}})  shape %[[VAL_6:.*]] : (!fir.class<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>, index, index, index, !fir.shape<1>) -> !fir.class<!fir.array<10x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>
  84 ! CHECK:  %[[VAL_8:.*]] = hlfir.parent_comp %[[VAL_7]] shape %[[VAL_6]] : (!fir.class<!fir.array<10x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>, !fir.shape<1>) -> !fir.box<!fir.array<10x!fir.type<_QMpc_typesTt{i:i32}>>>
  85 ! CHECK:  %[[VAL_9:.*]] = fir.convert %[[VAL_8]] : (!fir.box<!fir.array<10x!fir.type<_QMpc_typesTt{i:i32}>>>) -> !fir.box<!fir.array<?x!fir.type<_QMpc_typesTt{i:i32}>>>
  86 ! CHECK:  fir.call @_QPtakes_t_type_array(%[[VAL_9]])
  87
  88 subroutine test_pointer_leaf_parent_ref_array(x)
  89  use pc_types
  90  class(t2), pointer :: x(:)
  91  call takes_t_type_array(x%t)
  92 end subroutine
  93 ! CHECK-LABEL: func.func @_QPtest_pointer_leaf_parent_ref_array(
  94 ! CHECK:  %[[VAL_1:.*]]:2 = hlfir.declare {{.*}}Ex"
  95 ! CHECK:  %[[VAL_2:.*]] = fir.load %[[VAL_1]]#0 : !fir.ref<!fir.class<!fir.ptr<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>>>
  96 ! CHECK:  %[[VAL_3:.*]] = arith.constant 0 : index
  97 ! CHECK:  %[[VAL_4:.*]]:3 = fir.box_dims %[[VAL_2]], %[[VAL_3]] : (!fir.class<!fir.ptr<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>>, index) -> (index, index, index)
  98 ! CHECK:  %[[VAL_5:.*]] = fir.shape %[[VAL_4]]#1 : (index) -> !fir.shape<1>
  99 ! CHECK:  %[[VAL_6:.*]] = hlfir.parent_comp %[[VAL_2]] shape %[[VAL_5]] : (!fir.class<!fir.ptr<!fir.array<?x!fir.type<_QMpc_typesTt2{i:i32,j:i32}>>>>, !fir.shape<1>) -> !fir.box<!fir.array<?x!fir.type<_QMpc_typesTt{i:i32}>>>