flang/test/HLFIR/maxloc.fir

   1 // Test hlfir.maxloc operation parse, verify (no errors), and unparse
   2
   3 // RUN: fir-opt %s | fir-opt | FileCheck %s
   4
   5 // array is an expression of known shape
   6 func.func @maxloc0(%arg0: !hlfir.expr<42xi32>) {
   7   %mask = fir.alloca !fir.logical<4>
   8   %c_1 = arith.constant 1 : index
   9   %true = arith.constant true
  10   %true_logical = fir.convert %true : (i1) -> !fir.logical<4>
  11   fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
  12   %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
  13   %0 = hlfir.maxloc %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i32
  14   return
  15 }
  16 // CHECK:      func.func @maxloc0(%[[ARRAY:.*]]: !hlfir.expr<42xi32>) {
  17 // CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
  18 // CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
  19 // CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
  20 // CHECK-NEXT:   %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
  21 // CHECK-NEXT:   fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>
  22 // CHECK-NEXT:   %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
  23 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY]] dim %[[C1]] mask %[[BOX]] : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i32
  24 // CHECK-NEXT:   return
  25 // CHECK-NEXT: }
  26
  27 // array is an expression of assumed shape
  28 func.func @maxloc1(%arg0: !hlfir.expr<?xi32>) {
  29   %mask = fir.alloca !fir.logical<4>
  30   %c_1 = arith.constant 1 : index
  31   %true = arith.constant true
  32   %true_logical = fir.convert %true : (i1) -> !fir.logical<4>
  33   fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
  34   %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
  35   %0 = hlfir.maxloc %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i32
  36   return
  37 }
  38 // CHECK:      func.func @maxloc1(%[[ARRAY:.*]]: !hlfir.expr<?xi32>) {
  39 // CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
  40 // CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
  41 // CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
  42 // CHECK-NEXT:   %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
  43 // CHECK-NEXT:   fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>
  44 // CHECK-NEXT:   %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
  45 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i32
  46 // CHECK-NEXT:   return
  47 // CHECK-NEXT: }
  48
  49 // boxed array
  50 func.func @maxloc2(%arg0: !fir.box<!fir.array<42xi32>>) {
  51   %mask = fir.alloca !fir.logical<4>
  52   %c_1 = arith.constant 1 : index
  53   %true = arith.constant true
  54   %true_logical = fir.convert %true : (i1) -> !fir.logical<4>
  55   fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
  56   %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
  57   %0 = hlfir.maxloc %arg0 dim %c_1 mask %mask_box : (!fir.box<!fir.array<42xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
  58   return
  59 }
  60 // CHECK:      func.func @maxloc2(%[[ARRAY:.*]]: !fir.box<!fir.array<42xi32>>) {
  61 // CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
  62 // CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
  63 // CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
  64 // CHECK-NEXT:   %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
  65 // CHECK-NEXT:   fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>
  66 // CHECK-NEXT:   %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
  67 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!fir.box<!fir.array<42xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
  68 // CHECK-NEXT:   return
  69 // CHECK-NEXT: }
  70
  71 // assumed shape boxed array
  72 func.func @maxloc3(%arg0: !fir.box<!fir.array<?xi32>>) {
  73   %mask = fir.alloca !fir.logical<4>
  74   %c_1 = arith.constant 1 : index
  75   %true = arith.constant true
  76   %true_logical = fir.convert %true : (i1) -> !fir.logical<4>
  77   fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
  78   %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
  79   %0 = hlfir.maxloc %arg0 dim %c_1 mask %mask_box : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
  80   return
  81 }
  82 // CHECK:      func.func @maxloc3(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>) {
  83 // CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
  84 // CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
  85 // CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
  86 // CHECK-NEXT:   %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
  87 // CHECK-NEXT:   fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>
  88 // CHECK-NEXT:   %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
  89 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
  90 // CHECK-NEXT:   return
  91 // CHECK-NEXT: }
  92
  93 // known shape expr mask
  94 func.func @maxloc4(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<42x!fir.logical<4>>) {
  95   %c_1 = arith.constant 1 : index
  96   %0 = hlfir.maxloc %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<42x!fir.logical<4>>) -> i32
  97   return
  98 }
  99 // CHECK:      func.func @maxloc4(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !hlfir.expr<42x!fir.logical<4>>) {
 100 // CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
 101 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<42x!fir.logical<4>>) -> i32
 102 // CHECK-NEXT:   return
 103 // CHECK-NEXT: }
 104
 105 // assumed shape expr mask
 106 func.func @maxloc5(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<?x!fir.logical<4>>) {
 107   %c_1 = arith.constant 1 : index
 108   %0 = hlfir.maxloc %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<?x!fir.logical<4>>) -> i32
 109   return
 110 }
 111 // CHECK:      func.func @maxloc5(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !hlfir.expr<?x!fir.logical<4>>) {
 112 // CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
 113 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<?x!fir.logical<4>>) -> i32
 114 // CHECK-NEXT:   return
 115 // CHECK-NEXT: }
 116
 117 // known shape array mask
 118 func.func @maxloc6(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<42x!fir.logical<4>>>) {
 119   %c_1 = arith.constant 1 : index
 120   %0 = hlfir.maxloc %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<42x!fir.logical<4>>>) -> i32
 121   return
 122 }
 123 // CHECK:      func.func @maxloc6(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.box<!fir.array<42x!fir.logical<4>>>) {
 124 // CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
 125 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<42x!fir.logical<4>>>) -> i32
 126 // CHECK-NEXT:   return
 127 // CHECK-NEXT: }
 128
 129 // assumed shape array mask
 130 func.func @maxloc7(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<?x!fir.logical<4>>>) {
 131   %c_1 = arith.constant 1 : index
 132   %0 = hlfir.maxloc %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<?x!fir.logical<4>>>) -> i32
 133   return
 134 }
 135 // CHECK:      func.func @maxloc7(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>) {
 136 // CHECK-NEXT:   %[[C1:.*]] = arith.constant 1 : index
 137 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<?x!fir.logical<4>>>) -> i32
 138 // CHECK-NEXT:   return
 139 // CHECK-NEXT: }
 140
 141 // known shape expr return
 142 func.func @maxloc8(%arg0: !fir.box<!fir.array<2x2xi32>>, %arg1: i32) {
 143   %mask = fir.alloca !fir.logical<4>
 144   %true = arith.constant true
 145   %true_logical = fir.convert %true : (i1) -> !fir.logical<4>
 146   fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
 147   %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
 148   %0 = hlfir.maxloc %arg0 dim %arg1 mask %mask_box : (!fir.box<!fir.array<2x2xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<2xi32>
 149   return
 150 }
 151 // CHECK:      func.func @maxloc8(%[[ARRAY:.*]]: !fir.box<!fir.array<2x2xi32>>, %[[DIM:.*]]: i32) {
 152 // CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
 153 // CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
 154 // CHECK-NEXT:   %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
 155 // CHECK-NEXT:   fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>
 156 // CHECK-NEXT:   %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
 157 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY]] dim %[[DIM]] mask %[[BOX]] : (!fir.box<!fir.array<2x2xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<2xi32>
 158 // CHECK-NEXT:   return
 159 // CHECK-NEXT: }
 160
 161 // assumed shape expr return
 162 func.func @maxloc9(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {
 163   %mask = fir.alloca !fir.logical<4>
 164   %true = arith.constant true
 165   %true_logical = fir.convert %true : (i1) -> !fir.logical<4>
 166   fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
 167   %mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
 168   %0 = hlfir.maxloc %arg0 dim %arg1 mask %mask_box : (!fir.box<!fir.array<?x?xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<?xi32>
 169   return
 170 }
 171 // CHECK:      func.func @maxloc9(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32) {
 172 // CHECK-NEXT:   %[[MASK:.*]] = fir.alloca !fir.logical<4>
 173 // CHECK-NEXT:   %[[TRUE:.*]] = arith.constant true
 174 // CHECK-NEXT:   %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
 175 // CHECK-NEXT:   fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>
 176 // CHECK-NEXT:   %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
 177 // CHECK-NEXT:   hlfir.maxloc %[[ARRAY]] dim %[[DIM]] mask %[[BOX]] : (!fir.box<!fir.array<?x?xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<?xi32>
 178 // CHECK-NEXT:   return
 179 // CHECK-NEXT: }
 180
 181 // hlfir.maxloc with only an array argument
 182 func.func @maxloc10(%arg0: !fir.box<!fir.array<?x?xi32>>) {
 183   %maxloc = hlfir.maxloc %arg0 : (!fir.box<!fir.array<?x?xi32>>) -> !hlfir.expr<1xi32>
 184   return
 185 }
 186 // CHECK:      func.func @maxloc10(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>
 187 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] : (!fir.box<!fir.array<?x?xi32>>) -> !hlfir.expr<1xi32>
 188 // CHECK-NEXT:   return
 189 // CHECK-NEXT: }
 190
 191 // hlfir.maxloc with only a character array argument
 192 func.func @maxloc11(%arg0: !fir.box<!fir.array<?x?x!fir.char<1,?>>>) {
 193   %maxloc = hlfir.maxloc %arg0 : (!fir.box<!fir.array<?x?x!fir.char<1,?>>>) -> !hlfir.expr<1xi32>
 194   return
 195 }
 196 // CHECK:      func.func @maxloc11(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?x!fir.char<1,?>>>
 197 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] : (!fir.box<!fir.array<?x?x!fir.char<1,?>>>) -> !hlfir.expr<1xi32>
 198 // CHECK-NEXT:   return
 199 // CHECK-NEXT: }
 200
 201 // hlfir.maxloc with array and dim argument
 202 func.func @maxloc12(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {
 203   %maxloc = hlfir.maxloc %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>
 204   return
 205 }
 206 // CHECK:      func.func @maxloc12(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32
 207 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>
 208 // CHECK-NEXT:   return
 209 // CHECK-NEXT: }
 210
 211 // hlfir.maxloc with array and mask argument
 212 func.func @maxloc13(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.logical<4>) {
 213   %maxloc = hlfir.maxloc %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> !hlfir.expr<1xi32>
 214   return
 215 }
 216 // CHECK:      func.func @maxloc13(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.logical<4>
 217 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> !hlfir.expr<1xi32>
 218 // CHECK-NEXT:   return
 219 // CHECK-NEXT: }
 220
 221 // hlfir.maxloc with dim argument with an unusual type
 222 func.func @maxloc14(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: index) {
 223   %maxloc = hlfir.maxloc %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>
 224   return
 225 }
 226 // CHECK:      func.func @maxloc14(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: index
 227 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>
 228 // CHECK-NEXT:   return
 229 // CHECK-NEXT: }
 230
 231 // hlfir.maxloc with mask argument of unusual type
 232 func.func @maxloc15(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: i1) {
 233   %maxloc = hlfir.maxloc %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, i1) -> !hlfir.expr<1xi32>
 234   return
 235 }
 236 // CHECK:      func.func @maxloc15(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: i1
 237 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, i1) -> !hlfir.expr<1xi32>
 238 // CHECK-NEXT:   return
 239 // CHECK-NEXT: }
 240
 241 // hlfir.maxloc with mask argument of ref<array<>> type
 242 func.func @maxloc16(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.ref<!fir.array<?x!fir.logical<4>>>) {
 243   %maxloc = hlfir.maxloc %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> !hlfir.expr<1xi32>
 244   return
 245 }
 246 // CHECK:      func.func @maxloc16(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.ref<!fir.array<?x!fir.logical<4>>>
 247 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> !hlfir.expr<1xi32>
 248 // CHECK-NEXT:   return
 249 // CHECK-NEXT: }
 250
 251
 252 // hlfir.maxloc with kind implied by the return type
 253 func.func @maxloc17(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: i1) {
 254   %maxloc = hlfir.maxloc %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, i1) -> !hlfir.expr<1xi16>
 255   return
 256 }
 257 // CHECK:      func.func @maxloc17(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: i1
 258 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, i1) -> !hlfir.expr<1xi16>
 259 // CHECK-NEXT:   return
 260 // CHECK-NEXT: }
 261
 262 // hlfir.maxloc with back argument
 263 func.func @maxloc18(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: i1) {
 264   %true = arith.constant true
 265   %maxloc = hlfir.maxloc %arg0 mask %arg1 back %true : (!fir.box<!fir.array<?xi32>>, i1, i1) -> !hlfir.expr<1xi32>
 266   return
 267 }
 268 // CHECK:      func.func @maxloc18(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: i1
 269 // CHECK-NEXT:   %[[C2:.*]] = arith.constant true
 270 // CHECK-NEXT:   %[[maxloc:.*]] = hlfir.maxloc %[[ARRAY]] mask %[[MASK]] back %[[C2]] : (!fir.box<!fir.array<?xi32>>, i1, i1) -> !hlfir.expr<1xi32>
 271 // CHECK-NEXT:   return
 272 // CHECK-NEXT: }