flang/test/Lower/io-write.f90

   1 ! RUN: bbc -emit-fir -hlfir=false %s -o - | FileCheck %s
   2
   3 ! Test that IO item calls stackrestore in the right place
   4
   5 ! CHECK-LABEL: func.func @_QQmain() {
   6   character(3) string
   7   write(string,getstring(6))
   8 ! CHECK:  %[[Val_0:.*]] = fir.alloca i32 {adapt.valuebyref}
   9 ! CHECK:  %[[Const_3:.*]] = arith.constant 3 : index
  10 ! CHECK:  %[[Val_1:.*]] = fir.alloca !fir.char<1,3> {bindc_name = "string", uniq_name = "_QFEstring"}
  11 ! CHECK:  %[[Val_2:.*]] = fir.convert %[[Val_1]] : (!fir.ref<!fir.char<1,3>>) -> !fir.ref<i8>
  12 ! CHECK:  %[[Val_3:.*]] = fir.convert %[[Const_3]] : (index) -> i64
  13 ! CHECK:  %[[Const_6:.*]] = arith.constant 6 : i32
  14 ! CHECK:  fir.store %[[Const_6]] to %[[Val_0]] : !fir.ref<i32>
  15 ! CHECK:  %[[Val_4:.*]] = fir.load %[[Val_0]] : !fir.ref<i32>
  16 ! CHECK:  %[[Val_5:.*]] = fir.convert %[[Val_4]] : (i32) -> i64
  17 ! CHECK:  %[[Val_6:.*]] = fir.convert %[[Val_5]] : (i64) -> index
  18 ! CHECK:  %[[Const_0:.*]] = arith.constant 0 : index
  19 ! CHECK:  %[[Val_7:.*]] = arith.cmpi sgt, %[[Val_6]], %[[Const_0]] : index
  20 ! CHECK:  %[[Val_8:.*]] = arith.select %[[Val_7]], %[[Val_6]], %[[Const_0]] : index
  21 ! CHECK:  %[[Val_9:.*]] = fir.call @llvm.stacksave.p0() {{.*}}: () -> !fir.ref<i8>
  22 ! CHECK:  %[[Val_10:.*]] = fir.alloca !fir.char<1,?>(%[[Val_8]] : index) {bindc_name = ".result"}
  23 ! CHECK:  %[[Val_11:.*]] = fir.call @_QFPgetstring(%[[Val_10]], %[[Val_8]], %[[Val_0]]) {{.*}}: (!fir.ref<!fir.char<1,?>>, index, !fir.ref<i32>) -> !fir.boxchar<1>
  24 ! CHECK:  %[[Val_12:.*]] = fir.convert %[[Val_10]] : (!fir.ref<!fir.char<1,?>>) -> !fir.ref<i8>
  25 ! CHECK:  %[[Val_13:.*]] = fir.convert %[[Val_8]] : (index) -> i64
  26 ! CHECK:  %[[Val_14:.*]] = fir.zero_bits !fir.box<none>
  27 ! CHECK:  %[[Const_0_i64:.*]] = arith.constant 0 : i64
  28 ! CHECK:  %[[Val_15:.*]] = fir.convert %[[Const_0_i64]] : (i64) -> !fir.ref<!fir.llvm_ptr<i8>>
  29 ! CHECK:  %[[Const_0_i64_0:.*]] = arith.constant 0 : i64
  30 ! CHECK:  %[[Val_16:.*]] = fir.address_of(@_QQclX{{.*}}) : !fir.ref<!fir.char<1,{{.*}}>>
  31 ! CHECK:  %[[Val_17:.*]] = fir.convert %[[Val_16]] : (!fir.ref<!fir.char<1,{{.*}}>>) -> !fir.ref<i8>
  32 ! CHECK:  %[[Val_18:.*]] = fir.call @_FortranAioBeginInternalFormattedOutput(%[[Val_2]], %[[Val_3]], %[[Val_12]], %[[Val_13]],
  33 ! %[[Val_14]], %[[Val_15]], %[[Const_0_i64_0]], %17, %{{.*}}) : (!fir.ref<i8>, i64, !fir.ref<i8>, i64, !fir.box<none>, !fir.ref<!fir.llvm_ptr<i8>>, i64, !fir.ref<i8>, i32) -> !fir.ref<i8>
  34 ! CHECK:  %[[Val_19:.*]] = fir.call @_FortranAioEndIoStatement(%18) {{.*}}: (!fir.ref<i8>) -> i32
  35 ! CHECK:  fir.call @llvm.stackrestore.p0(%[[Val_9]]) {{.*}}: (!fir.ref<i8>) -> ()
  36   if (string/="hi") stop 'FAIL'
  37 contains
  38   function getstring(n) result(r)
  39     character(n) r
  40     r = '("hi")'
  41   end function
  42 end