flang/test/HLFIR/all.fir

   1 // Test hlfir.all operation parse, verify (no errors), and unparse
   2
   3 // RUN: fir-opt %s | fir-opt | FileCheck %s
   4
   5 // mask is an expression of known shape
   6 func.func @all0(%arg0: !hlfir.expr<2x!fir.logical<4>>) {
   7   %all = hlfir.all %arg0 : (!hlfir.expr<2x!fir.logical<4>>) -> !fir.logical<4>
   8   return
   9 }
  10 // CHECK:      func.func @all0(%[[ARRAY:.*]]: !hlfir.expr<2x!fir.logical<4>>) {
  11 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] : (!hlfir.expr<2x!fir.logical<4>>) -> !fir.logical<4>
  12 // CHECK-NEXT:   return
  13 // CHECK-NEXT: }
  14
  15 // mask is an expression of assumed shape
  16 func.func @all1(%arg0: !hlfir.expr<?x!fir.logical<4>>) {
  17   %all = hlfir.all %arg0 : (!hlfir.expr<?x!fir.logical<4>>) -> !fir.logical<4>
  18   return
  19 }
  20 // CHECK:      func.func @all1(%[[ARRAY:.*]]: !hlfir.expr<?x!fir.logical<4>>) {
  21 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] : (!hlfir.expr<?x!fir.logical<4>>) -> !fir.logical<4>
  22 // CHECK-NEXT:   return
  23 // CHECK-NEXT: }
  24
  25 // mask is a boxed array
  26 func.func @all2(%arg0: !fir.box<!fir.array<2x!fir.logical<4>>>) {
  27   %all = hlfir.all %arg0 : (!fir.box<!fir.array<2x!fir.logical<4>>>) -> !fir.logical<4>
  28   return
  29 }
  30 // CHECK:      func.func @all2(%[[ARRAY:.*]]: !fir.box<!fir.array<2x!fir.logical<4>>>) {
  31 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] : (!fir.box<!fir.array<2x!fir.logical<4>>>) -> !fir.logical<4>
  32 // CHECK-NEXT:   return
  33 // CHECK-NEXT: }
  34
  35 // mask is an assumed shape boxed array
  36 func.func @all3(%arg0: !fir.box<!fir.array<?x!fir.logical<4>>>){
  37   %all = hlfir.all %arg0 : (!fir.box<!fir.array<?x!fir.logical<4>>>) -> !fir.logical<4>
  38   return
  39 }
  40 // CHECK:      func.func @all3(%[[ARRAY:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>) {
  41 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] : (!fir.box<!fir.array<?x!fir.logical<4>>>) -> !fir.logical<4>
  42 // CHECK-NEXT:   return
  43 // CHECK-NEXT: }
  44
  45 // mask is a 2-dimensional array
  46 func.func @all4(%arg0: !fir.box<!fir.array<?x?x!fir.logical<4>>>){
  47   %all = hlfir.all %arg0 : (!fir.box<!fir.array<?x?x!fir.logical<4>>>) -> !fir.logical<4>
  48   return
  49 }
  50 // CHECK:      func.func @all4(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?x!fir.logical<4>>>) {
  51 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] : (!fir.box<!fir.array<?x?x!fir.logical<4>>>) -> !fir.logical<4>
  52 // CHECK-NEXT:   return
  53 // CHECK-NEXT: }
  54
  55 // mask and dim argument
  56 func.func @all5(%arg0: !fir.box<!fir.array<?x!fir.logical<4>>>, %arg1: i32) {
  57   %all = hlfir.all %arg0 dim %arg1 : (!fir.box<!fir.array<?x!fir.logical<4>>>, i32) -> !fir.logical<4>
  58   return
  59 }
  60 // CHECK:      func.func @all5(%[[ARRAY:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>, %[[DIM:.*]]: i32) {
  61 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x!fir.logical<4>>>, i32) -> !fir.logical<4>
  62 // CHECK-NEXT:   return
  63 // CHECK-NEXT: }
  64
  65 // hlfir.all with dim argument with an unusual type
  66 func.func @all6(%arg0: !fir.box<!fir.array<?x!fir.logical<4>>>, %arg1: index) {
  67   %all = hlfir.all %arg0 dim %arg1 : (!fir.box<!fir.array<?x!fir.logical<4>>>, index) ->!fir.logical<4>
  68   return
  69 }
  70 // CHECK:      func.func @all6(%[[ARRAY:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>, %[[DIM:.*]]: index) {
  71 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x!fir.logical<4>>>, index) -> !fir.logical<4>
  72 // CHECK-NEXT:   return
  73 // CHECK-NEXT: }
  74
  75 // mask is a 2 dimensional array with dim
  76 func.func @all7(%arg0: !fir.box<!fir.array<?x?x!fir.logical<4>>>, %arg1: i32) {
  77   %all = hlfir.all %arg0 dim %arg1 : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i32) -> !hlfir.expr<?x!fir.logical<4>>
  78   return
  79 }
  80 // CHECK:      func.func @all7(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?x!fir.logical<4>>>, %[[DIM:.*]]: i32) {
  81 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?x!fir.logical<4>>>, i32) -> !hlfir.expr<?x!fir.logical<4>>
  82 // CHECK-NEXT:   return
  83 // CHECK-NEXT: }
  84
  85 // known shape expr return
  86 func.func @all8(%arg0: !fir.box<!fir.array<2x2x!fir.logical<4>>>, %arg1: i32) {
  87   %all = hlfir.all %arg0 dim %arg1 : (!fir.box<!fir.array<2x2x!fir.logical<4>>>, i32) -> !hlfir.expr<2x!fir.logical<4>>
  88   return
  89 }
  90 // CHECK:      func.func @all8(%[[ARRAY:.*]]: !fir.box<!fir.array<2x2x!fir.logical<4>>>, %[[DIM:.*]]: i32) {
  91 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<2x2x!fir.logical<4>>>, i32) -> !hlfir.expr<2x!fir.logical<4>>
  92 // CHECK-NEXT:   return
  93 // CHECK-NEXT: }
  94
  95 // hlfir.all with mask argument of ref<array<>> type
  96 func.func @all9(%arg0: !fir.ref<!fir.array<?x!fir.logical<4>>>) {
  97   %all = hlfir.all %arg0 : (!fir.ref<!fir.array<?x!fir.logical<4>>>) -> !fir.logical<4>
  98   return
  99 }
 100 // CHECK:      func.func @all9(%[[ARRAY:.*]]: !fir.ref<!fir.array<?x!fir.logical<4>>>) {
 101 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] : (!fir.ref<!fir.array<?x!fir.logical<4>>>) -> !fir.logical<4>
 102 // CHECK-NEXT:   return
 103 // CHECK-NEXT: }
 104
 105 // hlfir.all with fir.logical<8> type
 106 func.func @all10(%arg0: !fir.box<!fir.array<?x!fir.logical<8>>>) {
 107   %all = hlfir.all %arg0 : (!fir.box<!fir.array<?x!fir.logical<8>>>) -> !fir.logical<8>
 108   return
 109 }
 110 // CHECK:      func.func @all10(%[[ARRAY:.*]]: !fir.box<!fir.array<?x!fir.logical<8>>>) {
 111 // CHECK-NEXT:   %[[ALL:.*]] = hlfir.all %[[ARRAY]] : (!fir.box<!fir.array<?x!fir.logical<8>>>) -> !fir.logical<8>
 112 // CHECK-NEXT:   return
 113 // CHECK-NEXT: }