flang/test/Lower/HLFIR/vector-subscript-lhs.f90

   1 ! Test lowering of vector subscripted designators in assignment
   2 ! left-hand sides.
   3 ! RUN: bbc -emit-hlfir -o - -I nw %s 2>&1 | FileCheck %s
   4
   5 subroutine test_simple(x, vector)
   6   integer(8) :: vector(10)
   7   real :: x(:)
   8   x(vector) = 42.
   9 end subroutine
  10 ! CHECK-LABEL:   func.func @_QPtest_simple(
  11 ! CHECK:  %[[VAL_4:.*]]:2 = hlfir.declare {{.*}}Evector
  12 ! CHECK:  %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}Ex
  13 ! CHECK:  hlfir.region_assign {
  14 ! CHECK:    %[[VAL_6:.*]] = arith.constant 4.200000e+01 : f32
  15 ! CHECK:    hlfir.yield %[[VAL_6]] : f32
  16 ! CHECK:  } to {
  17 ! CHECK:    %[[VAL_7:.*]] = arith.constant 10 : index
  18 ! CHECK:    %[[VAL_8:.*]] = fir.shape %[[VAL_7]] : (index) -> !fir.shape<1>
  19 ! CHECK:    hlfir.elemental_addr %[[VAL_8]] unordered : !fir.shape<1> {
  20 ! CHECK:    ^bb0(%[[VAL_9:.*]]: index):
  21 ! CHECK:      %[[VAL_10:.*]] = hlfir.designate %[[VAL_4]]#0 (%[[VAL_9]])  : (!fir.ref<!fir.array<10xi64>>, index) -> !fir.ref<i64>
  22 ! CHECK:      %[[VAL_11:.*]] = fir.load %[[VAL_10]] : !fir.ref<i64>
  23 ! CHECK:      %[[VAL_12:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_11]])  : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>
  24 ! CHECK:      hlfir.yield %[[VAL_12]] : !fir.ref<f32>
  25 ! CHECK:    }
  26 ! CHECK:  }
  27
  28 subroutine test_cleanup(x, vector, matrix)
  29   integer(8) :: vector(10), matrix(10, 5)
  30   real :: x(:)
  31   x(matmul(vector, matrix)) = 42.
  32 end subroutine
  33 ! CHECK-LABEL:   func.func @_QPtest_cleanup(
  34 ! CHECK:  %[[VAL_6:.*]]:2 = hlfir.declare {{.*}}Ematrix
  35 ! CHECK:  %[[VAL_9:.*]]:2 = hlfir.declare {{.*}}Evector
  36 ! CHECK:  %[[VAL_10:.*]]:2 = hlfir.declare {{.*}}Ex
  37 ! CHECK:  hlfir.region_assign {
  38 ! CHECK:    %[[VAL_11:.*]] = arith.constant 4.200000e+01 : f32
  39 ! CHECK:    hlfir.yield %[[VAL_11]] : f32
  40 ! CHECK:  } to {
  41 ! CHECK:    %[[VAL_12:.*]] = hlfir.matmul %[[VAL_9]]#0 %[[VAL_6]]#0 {fastmath = #arith.fastmath<contract>} : (!fir.ref<!fir.array<10xi64>>, !fir.ref<!fir.array<10x5xi64>>) -> !hlfir.expr<5xi64>
  42 ! CHECK:    %[[VAL_13:.*]] = arith.constant 5 : index
  43 ! CHECK:    %[[VAL_14:.*]] = fir.shape %[[VAL_13]] : (index) -> !fir.shape<1>
  44 ! CHECK:    hlfir.elemental_addr %[[VAL_14]] unordered : !fir.shape<1> {
  45 ! CHECK:    ^bb0(%[[VAL_15:.*]]: index):
  46 ! CHECK:      %[[VAL_16:.*]] = hlfir.apply %[[VAL_12]], %[[VAL_15]] : (!hlfir.expr<5xi64>, index) -> i64
  47 ! CHECK:      %[[VAL_17:.*]] = hlfir.designate %[[VAL_10]]#0 (%[[VAL_16]])  : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>
  48 ! CHECK:      hlfir.yield %[[VAL_17]] : !fir.ref<f32>
  49 ! CHECK:    } cleanup {
  50 ! CHECK:      hlfir.destroy %[[VAL_12]] : !hlfir.expr<5xi64>
  51 ! CHECK:    }
  52 ! CHECK:  }
  53
  54 subroutine test_nested_vectors(x, vector1, vector2, vector3)
  55   integer(8) :: vector1(10), vector2(8), vector3(6)
  56   real :: x(:)
  57   x(vector1(vector2(vector3))) = 42.
  58 end subroutine
  59 ! CHECK-LABEL:   func.func @_QPtest_nested_vectors(
  60 ! CHECK:  %[[VAL_6:.*]]:2 = hlfir.declare {{.*}}Evector1
  61 ! CHECK:  %[[VAL_9:.*]]:2 = hlfir.declare {{.*}}Evector2
  62 ! CHECK:  %[[VAL_12:.*]]:2 = hlfir.declare {{.*}}Evector3
  63 ! CHECK:  %[[VAL_13:.*]]:2 = hlfir.declare {{.*}}Ex
  64 ! CHECK:  hlfir.region_assign {
  65 ! CHECK:    %[[VAL_14:.*]] = arith.constant 4.200000e+01 : f32
  66 ! CHECK:    hlfir.yield %[[VAL_14]] : f32
  67 ! CHECK:  } to {
  68 ! CHECK:    %[[VAL_15:.*]] = arith.constant 6 : index
  69 ! CHECK:    %[[VAL_16:.*]] = fir.shape %[[VAL_15]] : (index) -> !fir.shape<1>
  70 ! CHECK:    %[[VAL_17:.*]] = hlfir.elemental %[[VAL_16]] unordered : (!fir.shape<1>) -> !hlfir.expr<6xi64> {
  71 ! CHECK:    ^bb0(%[[VAL_18:.*]]: index):
  72 ! CHECK:      %[[VAL_19:.*]] = hlfir.designate %[[VAL_12]]#0 (%[[VAL_18]])  : (!fir.ref<!fir.array<6xi64>>, index) -> !fir.ref<i64>
  73 ! CHECK:      %[[VAL_20:.*]] = fir.load %[[VAL_19]] : !fir.ref<i64>
  74 ! CHECK:      %[[VAL_21:.*]] = hlfir.designate %[[VAL_9]]#0 (%[[VAL_20]])  : (!fir.ref<!fir.array<8xi64>>, i64) -> !fir.ref<i64>
  75 ! CHECK:      %[[VAL_22:.*]] = fir.load %[[VAL_21]] : !fir.ref<i64>
  76 ! CHECK:      hlfir.yield_element %[[VAL_22]] : i64
  77 ! CHECK:    }
  78 ! CHECK:    %[[VAL_23:.*]] = arith.constant 6 : index
  79 ! CHECK:    %[[VAL_24:.*]] = fir.shape %[[VAL_23]] : (index) -> !fir.shape<1>
  80 ! CHECK:    %[[VAL_25:.*]] = hlfir.elemental %[[VAL_24]] unordered : (!fir.shape<1>) -> !hlfir.expr<6xi64> {
  81 ! CHECK:    ^bb0(%[[VAL_26:.*]]: index):
  82 ! CHECK:      %[[VAL_27:.*]] = hlfir.apply %[[VAL_28:.*]], %[[VAL_26]] : (!hlfir.expr<6xi64>, index) -> i64
  83 ! CHECK:      %[[VAL_29:.*]] = hlfir.designate %[[VAL_6]]#0 (%[[VAL_27]])  : (!fir.ref<!fir.array<10xi64>>, i64) -> !fir.ref<i64>
  84 ! CHECK:      %[[VAL_30:.*]] = fir.load %[[VAL_29]] : !fir.ref<i64>
  85 ! CHECK:      hlfir.yield_element %[[VAL_30]] : i64
  86 ! CHECK:    }
  87 ! CHECK:    %[[VAL_31:.*]] = arith.constant 6 : index
  88 ! CHECK:    %[[VAL_32:.*]] = fir.shape %[[VAL_31]] : (index) -> !fir.shape<1>
  89 ! CHECK:    hlfir.elemental_addr %[[VAL_32]] unordered : !fir.shape<1> {
  90 ! CHECK:    ^bb0(%[[VAL_33:.*]]: index):
  91 ! CHECK:      %[[VAL_34:.*]] = hlfir.apply %[[VAL_35:.*]], %[[VAL_33]] : (!hlfir.expr<6xi64>, index) -> i64
  92 ! CHECK:      %[[VAL_36:.*]] = hlfir.designate %[[VAL_13]]#0 (%[[VAL_34]])  : (!fir.box<!fir.array<?xf32>>, i64) -> !fir.ref<f32>
  93 ! CHECK:      hlfir.yield %[[VAL_36]] : !fir.ref<f32>
  94 ! CHECK:    } cleanup {
  95 ! CHECK:      hlfir.destroy %[[VAL_37:.*]] : !hlfir.expr<6xi64>
  96 ! CHECK:      hlfir.destroy %[[VAL_38:.*]] : !hlfir.expr<6xi64>
  97 ! CHECK:    }
  98 ! CHECK:  }
  99
 100 subroutine test_substring(x, vector)
 101   integer(8) :: vector(10), ifoo, ibar
 102   external :: ifoo, ibar
 103   character(*) :: x(:)
 104   x(vector)(ifoo(): ibar()) = "hello"
 105 end subroutine
 106 ! CHECK-LABEL:   func.func @_QPtest_substring(
 107 ! CHECK:  %[[VAL_4:.*]]:2 = hlfir.declare {{.*}}Evector
 108 ! CHECK:  %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}Ex
 109 ! CHECK:  hlfir.region_assign {
 110 ! CHECK:    %[[VAL_6:.*]] = fir.address_of(@{{.*}}) : !fir.ref<!fir.char<1,5>>
 111 ! CHECK:    %[[VAL_7:.*]] = arith.constant 5 : index
 112 ! CHECK:    %[[VAL_8:.*]]:2 = hlfir.declare %[[VAL_6]] typeparams %[[VAL_7]] {fortran_attrs = #fir.var_attrs<parameter>, uniq_name = "_QQclX68656C6C6F"} : (!fir.ref<!fir.char<1,5>>, index) -> (!fir.ref<!fir.char<1,5>>, !fir.ref<!fir.char<1,5>>)
 113 ! CHECK:    hlfir.yield %[[VAL_8]]#0 : !fir.ref<!fir.char<1,5>>
 114 ! CHECK:  } to {
 115 ! CHECK:    %[[VAL_9:.*]] = arith.constant 10 : index
 116 ! CHECK:    %[[VAL_10:.*]] = fir.shape %[[VAL_9]] : (index) -> !fir.shape<1>
 117 ! CHECK:    %[[VAL_11:.*]] = fir.call @_QPifoo() {{.*}}: () -> i64
 118 ! CHECK:    %[[VAL_12:.*]] = fir.call @_QPibar() {{.*}}: () -> i64
 119 ! CHECK:    %[[VAL_13:.*]] = fir.convert %[[VAL_11]] : (i64) -> index
 120 ! CHECK:    %[[VAL_14:.*]] = fir.convert %[[VAL_12]] : (i64) -> index
 121 ! CHECK:    %[[VAL_15:.*]] = arith.constant 1 : index
 122 ! CHECK:    %[[VAL_16:.*]] = arith.subi %[[VAL_14]], %[[VAL_13]] : index
 123 ! CHECK:    %[[VAL_17:.*]] = arith.addi %[[VAL_16]], %[[VAL_15]] : index
 124 ! CHECK:    %[[VAL_18:.*]] = arith.constant 0 : index
 125 ! CHECK:    %[[VAL_19:.*]] = arith.cmpi sgt, %[[VAL_17]], %[[VAL_18]] : index
 126 ! CHECK:    %[[VAL_20:.*]] = arith.select %[[VAL_19]], %[[VAL_17]], %[[VAL_18]] : index
 127 ! CHECK:    hlfir.elemental_addr %[[VAL_10]] typeparams %[[VAL_20]] unordered : !fir.shape<1>, index {
 128 ! CHECK:    ^bb0(%[[VAL_21:.*]]: index):
 129 ! CHECK:      %[[VAL_22:.*]] = hlfir.designate %[[VAL_4]]#0 (%[[VAL_21]])  : (!fir.ref<!fir.array<10xi64>>, index) -> !fir.ref<i64>
 130 ! CHECK:      %[[VAL_23:.*]] = fir.load %[[VAL_22]] : !fir.ref<i64>
 131 ! CHECK:      %[[VAL_24:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_23]]) substr %[[VAL_13]], %[[VAL_14]]  typeparams %[[VAL_20]] : (!fir.box<!fir.array<?x!fir.char<1,?>>>, i64, index, index, index) -> !fir.boxchar<1>
 132 ! CHECK:      hlfir.yield %[[VAL_24]] : !fir.boxchar<1>
 133 ! CHECK:    }
 134 ! CHECK:  }
 135
 136 subroutine test_hard_array_ref(x, vector1, vector2)
 137   integer(8) :: vector1(10), vector2(20), ifoo, ibar, ibaz
 138   external :: ifoo, ibar, ibaz
 139   real :: x(:, :, :, :, :)
 140   x(vector1, :, ifoo():ibar(), ibaz(), vector2) = 42.
 141 end subroutine
 142 ! CHECK-LABEL:   func.func @_QPtest_hard_array_ref(
 143 ! CHECK:  %[[VAL_5:.*]]:2 = hlfir.declare {{.*}}Evector1
 144 ! CHECK:  %[[VAL_8:.*]]:2 = hlfir.declare {{.*}}Evector2
 145 ! CHECK:  %[[VAL_9:.*]]:2 = hlfir.declare {{.*}}Ex
 146 ! CHECK:  hlfir.region_assign {
 147 ! CHECK:    %[[VAL_10:.*]] = arith.constant 4.200000e+01 : f32
 148 ! CHECK:    hlfir.yield %[[VAL_10]] : f32
 149 ! CHECK:  } to {
 150 ! CHECK:    %[[VAL_11:.*]] = arith.constant 10 : index
 151 ! CHECK:    %[[VAL_12:.*]] = arith.constant 1 : index
 152 ! CHECK:    %[[VAL_13:.*]] = arith.constant 1 : index
 153 ! CHECK:    %[[VAL_14:.*]]:3 = fir.box_dims %[[VAL_9]]#1, %[[VAL_13]] : (!fir.box<!fir.array<?x?x?x?x?xf32>>, index) -> (index, index, index)
 154 ! CHECK:    %[[VAL_15:.*]] = arith.constant 1 : index
 155 ! CHECK:    %[[VAL_16:.*]] = arith.constant 0 : index
 156 ! CHECK:    %[[VAL_17:.*]] = arith.subi %[[VAL_14]]#1, %[[VAL_12]] : index
 157 ! CHECK:    %[[VAL_18:.*]] = arith.addi %[[VAL_17]], %[[VAL_15]] : index
 158 ! CHECK:    %[[VAL_19:.*]] = arith.divsi %[[VAL_18]], %[[VAL_15]] : index
 159 ! CHECK:    %[[VAL_20:.*]] = arith.cmpi sgt, %[[VAL_19]], %[[VAL_16]] : index
 160 ! CHECK:    %[[VAL_21:.*]] = arith.select %[[VAL_20]], %[[VAL_19]], %[[VAL_16]] : index
 161 ! CHECK:    %[[VAL_22:.*]] = fir.call @_QPifoo() {{.*}}: () -> i64
 162 ! CHECK:    %[[VAL_23:.*]] = fir.call @_QPibar() {{.*}}: () -> i64
 163 ! CHECK:    %[[VAL_24:.*]] = fir.convert %[[VAL_22]] : (i64) -> index
 164 ! CHECK:    %[[VAL_25:.*]] = fir.convert %[[VAL_23]] : (i64) -> index
 165 ! CHECK:    %[[VAL_26:.*]] = arith.constant 1 : index
 166 ! CHECK:    %[[VAL_27:.*]] = arith.constant 0 : index
 167 ! CHECK:    %[[VAL_28:.*]] = arith.subi %[[VAL_25]], %[[VAL_24]] : index
 168 ! CHECK:    %[[VAL_29:.*]] = arith.addi %[[VAL_28]], %[[VAL_26]] : index
 169 ! CHECK:    %[[VAL_30:.*]] = arith.divsi %[[VAL_29]], %[[VAL_26]] : index
 170 ! CHECK:    %[[VAL_31:.*]] = arith.cmpi sgt, %[[VAL_30]], %[[VAL_27]] : index
 171 ! CHECK:    %[[VAL_32:.*]] = arith.select %[[VAL_31]], %[[VAL_30]], %[[VAL_27]] : index
 172 ! CHECK:    %[[VAL_33:.*]] = fir.call @_QPibaz() {{.*}}: () -> i64
 173 ! CHECK:    %[[VAL_34:.*]] = arith.constant 20 : index
 174 ! CHECK:    %[[VAL_35:.*]] = fir.shape %[[VAL_11]], %[[VAL_21]], %[[VAL_32]], %[[VAL_34]] : (index, index, index, index) -> !fir.shape<4>
 175 ! CHECK:    hlfir.elemental_addr %[[VAL_35]] unordered : !fir.shape<4> {
 176 ! CHECK:    ^bb0(%[[VAL_36:.*]]: index, %[[VAL_37:.*]]: index, %[[VAL_38:.*]]: index, %[[VAL_39:.*]]: index):
 177 ! CHECK:      %[[VAL_40:.*]] = hlfir.designate %[[VAL_5]]#0 (%[[VAL_36]])  : (!fir.ref<!fir.array<10xi64>>, index) -> !fir.ref<i64>
 178 ! CHECK:      %[[VAL_41:.*]] = fir.load %[[VAL_40]] : !fir.ref<i64>
 179 ! CHECK:      %[[VAL_42:.*]] = arith.constant 1 : index
 180 ! CHECK:      %[[VAL_43:.*]] = arith.subi %[[VAL_37]], %[[VAL_42]] : index
 181 ! CHECK:      %[[VAL_44:.*]] = arith.muli %[[VAL_43]], %[[VAL_15]] : index
 182 ! CHECK:      %[[VAL_45:.*]] = arith.addi %[[VAL_12]], %[[VAL_44]] : index
 183 ! CHECK:      %[[VAL_46:.*]] = arith.constant 1 : index
 184 ! CHECK:      %[[VAL_47:.*]] = arith.subi %[[VAL_38]], %[[VAL_46]] : index
 185 ! CHECK:      %[[VAL_48:.*]] = arith.muli %[[VAL_47]], %[[VAL_26]] : index
 186 ! CHECK:      %[[VAL_49:.*]] = arith.addi %[[VAL_24]], %[[VAL_48]] : index
 187 ! CHECK:      %[[VAL_50:.*]] = hlfir.designate %[[VAL_8]]#0 (%[[VAL_39]])  : (!fir.ref<!fir.array<20xi64>>, index) -> !fir.ref<i64>
 188 ! CHECK:      %[[VAL_51:.*]] = fir.load %[[VAL_50]] : !fir.ref<i64>
 189 ! CHECK:      %[[VAL_52:.*]] = hlfir.designate %[[VAL_9]]#0 (%[[VAL_41]], %[[VAL_45]], %[[VAL_49]], %[[VAL_33]], %[[VAL_51]])  : (!fir.box<!fir.array<?x?x?x?x?xf32>>, i64, index, index, i64, i64) -> !fir.ref<f32>
 190 ! CHECK:      hlfir.yield %[[VAL_52]] : !fir.ref<f32>
 191 ! CHECK:    }
 192 ! CHECK:  }
 193 ! CHECK:  return