flang/test/Lower/Intrinsics/scan.f90

   1 ! RUN: bbc -emit-fir %s -o - | FileCheck %s
   2
   3 ! CHECK-LABEL: func @_QPscan_test(
   4 ! CHECK-SAME: %[[s:[^:]+]]: !fir.boxchar<1>{{.*}}, %[[ss:[^:]+]]: !fir.boxchar<1>{{.*}}) -> i32
   5 integer function scan_test(s1, s2)
   6 character(*) :: s1, s2
   7 ! CHECK: %[[tmpBox:.*]] = fir.alloca !fir.box<!fir.heap<i32>>
   8 ! CHECK-DAG: %[[c:.*]]:2 = fir.unboxchar %[[s]] : (!fir.boxchar<1>) -> (!fir.ref<!fir.char<1,?>>, index)
   9 ! CHECK-DAG: %[[cBox:.*]] = fir.embox %[[c]]#0 typeparams %[[c]]#1 : (!fir.ref<!fir.char<1,?>>, index) -> !fir.box<!fir.char<1,?>>
  10 ! CHECK-DAG: %[[cBoxNone:.*]] = fir.convert %[[cBox]] : (!fir.box<!fir.char<1,?>>) -> !fir.box<none>
  11 ! CHECK-DAG: %[[c2:.*]]:2 = fir.unboxchar %[[ss]] : (!fir.boxchar<1>) -> (!fir.ref<!fir.char<1,?>>, index)
  12 ! CHECK-DAG: %[[cBox2:.*]] = fir.embox %[[c2]]#0 typeparams %[[c2]]#1 : (!fir.ref<!fir.char<1,?>>, index) -> !fir.box<!fir.char<1,?>>
  13 ! CHECK-DAG: %[[cBoxNone2:.*]] = fir.convert %[[cBox2]] : (!fir.box<!fir.char<1,?>>) -> !fir.box<none>
  14 ! CHECK-DAG: %[[backOptBox:.*]] = fir.absent !fir.box<i1>
  15 ! CHECK-DAG: %[[backBox:.*]] = fir.convert %[[backOptBox]] : (!fir.box<i1>) -> !fir.box<none>
  16 ! CHECK-DAG: %[[kindConstant:.*]] = arith.constant 4 : i32
  17 ! CHECK-DAG: %[[resBox:.*]] = fir.convert %[[tmpBox:.*]] : (!fir.ref<!fir.box<!fir.heap<i32>>>) -> !fir.ref<!fir.box<none>>
  18 ! CHECK: fir.call @{{.*}}Scan(%[[resBox]], %[[cBoxNone]], %[[cBoxNone2]], %[[backBox]], %[[kindConstant]], {{.*}}) {{.*}}: (!fir.ref<!fir.box<none>>, !fir.box<none>, !fir.box<none>, !fir.box<none>, i32, !fir.ref<i8>, i32) -> none
  19 scan_test = scan(s1, s2, kind=4)
  20 ! CHECK-DAG: %[[tmpAddr:.*]] = fir.box_addr
  21 ! CHECK: fir.freemem %[[tmpAddr]]
  22 end function scan_test
  23
  24 ! CHECK-LABEL: func @_QPscan_test2(
  25 ! CHECK-SAME: %[[s:[^:]+]]: !fir.boxchar<1>{{.*}},
  26 ! CHECK-SAME: %[[ss:[^:]+]]: !fir.boxchar<1>{{.*}}) -> i32
  27 integer function scan_test2(s1, s2)
  28 character(*) :: s1, s2
  29 ! CHECK: %[[st:[^:]*]]:2 = fir.unboxchar %[[s]] : (!fir.boxchar<1>) -> (!fir.ref<!fir.char<1,?>>, index)
  30 ! CHECK: %[[sst:[^:]*]]:2 = fir.unboxchar %[[ss]] : (!fir.boxchar<1>) -> (!fir.ref<!fir.char<1,?>>, index)
  31 ! CHECK: %[[a1:.*]] = fir.convert %[[st]]#0 : (!fir.ref<!fir.char<1,?>>) -> !fir.ref<i8>
  32 ! CHECK: %[[a2:.*]] = fir.convert %[[st]]#1 : (index) -> i64
  33 ! CHECK: %[[a3:.*]] = fir.convert %[[sst]]#0 : (!fir.ref<!fir.char<1,?>>) -> !fir.ref<i8>
  34 ! CHECK: %[[a4:.*]] = fir.convert %[[sst]]#1 : (index) -> i64
  35 ! CHECK: = fir.call @_FortranAScan1(%[[a1]], %[[a2]], %[[a3]], %[[a4]], %{{.*}}) {{.*}}: (!fir.ref<i8>, i64, !fir.ref<i8>, i64, i1) -> i64
  36 scan_test2 = scan(s1, s2, .true.)
  37 end function scan_test2
  38
  39 ! CHECK-LABEL: func @_QPtest_optional(
  40 ! CHECK-SAME:  %[[VAL_0:.*]]: !fir.box<!fir.array<?x!fir.char<1,?>>>
  41 ! CHECK-SAME:  %[[VAL_1:.*]]: !fir.boxchar<1>
  42 ! CHECK-SAME:  %[[VAL_2:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>
  43 subroutine test_optional(string, set, back)
  44 character (*) :: string(:), set
  45 logical, optional :: back(:)
  46 print *, scan(string, set, back)
  47 ! CHECK:  %[[VAL_11:.*]] = fir.is_present %[[VAL_2]] : (!fir.box<!fir.array<?x!fir.logical<4>>>) -> i1
  48 ! CHECK:  %[[VAL_12:.*]] = fir.zero_bits !fir.ref<!fir.array<?x!fir.logical<4>>>
  49 ! CHECK:  %[[VAL_13:.*]] = arith.constant 0 : index
  50 ! CHECK:  %[[VAL_14:.*]] = fir.shape %[[VAL_13]] : (index) -> !fir.shape<1>
  51 ! CHECK:  %[[VAL_15:.*]] = fir.embox %[[VAL_12]](%[[VAL_14]]) : (!fir.ref<!fir.array<?x!fir.logical<4>>>, !fir.shape<1>) -> !fir.box<!fir.array<?x!fir.logical<4>>>
  52 ! CHECK:  %[[VAL_16:.*]] = arith.select %[[VAL_11]], %[[VAL_2]], %[[VAL_15]] : !fir.box<!fir.array<?x!fir.logical<4>>>
  53 ! CHECK:  %[[VAL_17:.*]] = fir.array_load %[[VAL_16]] {fir.optional} : (!fir.box<!fir.array<?x!fir.logical<4>>>) -> !fir.array<?x!fir.logical<4>>
  54 ! CHECK:  fir.do_loop %[[VAL_25:.*]] = %{{.*}} to %{{.*}} step %{{.*}} unordered iter_args(%{{.*}} = %{{.*}}) -> (!fir.array<?xi32>) {
  55 ! CHECK:  %[[VAL_31:.*]] = fir.if %[[VAL_11]] -> (!fir.logical<4>) {
  56   ! CHECK:  %[[VAL_32:.*]] = fir.array_fetch %[[VAL_17]], %[[VAL_25]] : (!fir.array<?x!fir.logical<4>>, index) -> !fir.logical<4>
  57   ! CHECK:  fir.result %[[VAL_32]] : !fir.logical<4>
  58 ! CHECK:  } else {
  59   ! CHECK:  %[[VAL_33:.*]] = arith.constant false
  60   ! CHECK:  %[[VAL_34:.*]] = fir.convert %[[VAL_33]] : (i1) -> !fir.logical<4>
  61   ! CHECK:  fir.result %[[VAL_34]] : !fir.logical<4>
  62 ! CHECK:  }
  63 ! CHECK:  %[[VAL_39:.*]] = fir.convert %[[VAL_31]] : (!fir.logical<4>) -> i1
  64 ! CHECK:  fir.call @_FortranAScan1(%{{.*}}, %{{.*}}, %{{.*}}, %{{.*}}, %[[VAL_39]]) {{.*}}: (!fir.ref<i8>, i64, !fir.ref<i8>, i64, i1) -> i64
  65 ! CHECK:  }
  66 ! CHECK:  fir.array_merge_store
  67 end subroutine
  68
  69 ! CHECK-LABEL: func @_QPtest_optional_scalar(
  70 ! CHECK-SAME:  %[[VAL_0:.*]]: !fir.box<!fir.array<?x!fir.char<1,?>>>
  71 ! CHECK-SAME:  %[[VAL_1:.*]]: !fir.boxchar<1>
  72 ! CHECK-SAME:  %[[VAL_2:.*]]: !fir.ref<!fir.logical<4>>
  73 subroutine test_optional_scalar(string, set, back)
  74 character (*) :: string(:), set
  75 logical, optional :: back
  76 print *, scan(string, set, back)
  77 ! CHECK:  %[[VAL_11:.*]] = fir.is_present %[[VAL_2]] : (!fir.ref<!fir.logical<4>>) -> i1
  78 ! CHECK:  %[[VAL_12:.*]] = fir.if %[[VAL_11]] -> (!fir.logical<4>) {
  79 ! CHECK:  %[[VAL_13:.*]] = fir.load %[[VAL_2]] : !fir.ref<!fir.logical<4>>
  80 ! CHECK:  fir.result %[[VAL_13]] : !fir.logical<4>
  81 ! CHECK:  } else {
  82 ! CHECK:  %[[VAL_14:.*]] = arith.constant false
  83 ! CHECK:  %[[VAL_15:.*]] = fir.convert %[[VAL_14]] : (i1) -> !fir.logical<4>
  84 ! CHECK:  fir.result %[[VAL_15]] : !fir.logical<4>
  85 ! CHECK:  }
  86 ! CHECK:  fir.do_loop %{{.*}} = %{{.*}} to %{{.*}} step %{{.*}} unordered iter_args(%{{.*}} = %{{.*}}) -> (!fir.array<?xi32>) {
  87 ! CHECK:  %[[VAL_39:.*]] = fir.convert %[[VAL_12]] : (!fir.logical<4>) -> i1
  88 ! CHECK:  fir.call @_FortranAScan1(%{{.*}}, %{{.*}}, %{{.*}}, %{{.*}}, %[[VAL_39]]) {{.*}}: (!fir.ref<i8>, i64, !fir.ref<i8>, i64, i1) -> i64
  89 ! CHECK:  }
  90 ! CHECK:  fir.array_merge_store
  91 end subroutine