[flang] Add IO lowering test

This patch adds more lowering tests for IO
Test lowering of IO read SIZE control-spec (12.6.2.15)

This patch is part of the upstreaming effort from fir-dev branch.

Reviewed By: jeanPerier

Differential Revision: https://reviews.llvm.org/D121920

Co-authored-by: V Donaldson <vdonaldson@nvidia.com>

diff --git a/flang/test/Lower/io-statement-3.f90 b/flang/test/Lower/io-statement-3.f90
new file mode 100644 (file)
index 0000000..0d02271
--- /dev/null
+++ b/flang/test/Lower/io-statement-3.f90
@@ -0,0 +1,49 @@
+! Test lowering of IO read SIZE control-spec (12.6.2.15)
+! RUN: bbc -emit-fir -o - %s | FileCheck %s
+
+! CHECK-LABEL: func @_QPtest_read_size(
+! CHECK-SAME: %[[sizeVar:[^:]+]]: !fir.ref<i32>{{[^,]*}},
+subroutine test_read_size(size, c1, c2, unit, stat)
+  integer :: unit, size, stat
+  character(*) :: c1, c2
+  ! CHECK: %[[cookie:.*]] = fir.call @_FortranAioBeginExternalFormattedInput(
+  ! CHECK: fir.call @_FortranAioEnableHandlers(
+  ! CHECK: %[[ok1:.*]] = fir.call @_FortranAioSetAdvance(
+  ! CHECK: fir.if %[[ok1]] {
+  ! CHECK:   fir.if %[[ok1]] {
+  ! CHECK:     %[[ok2:.*]] = fir.call @_FortranAioInputAscii(
+  ! CHECK:     fir.if %[[ok2]] {
+  ! CHECK:       fir.call @_FortranAioInputAscii(
+  ! CHECK:     }
+  ! CHECK:   }
+  ! CHECK: }
+  ! CHECK: %[[sizeValue:.*]] = fir.call @_FortranAioGetSize(%[[cookie]]) : (!fir.ref<i8>) -> i64
+  ! CHECK: %[[sizeCast:.*]] = fir.convert %[[sizeValue]] : (i64) -> i32
+  ! CHECK: fir.store %[[sizeCast]] to %[[sizeVar]] : !fir.ref<i32>
+  ! CHECK: fir.call @_FortranAioEndIoStatement(%[[cookie]]) : (!fir.ref<i8>) -> i32
+  READ(unit, '(A)', ADVANCE='NO', SIZE=size, IOSTAT=stat) c1, c2
+end subroutine
+
+  ! CHECK: %[[unit:.*]] = fir.alloca i32 {bindc_name = "unit", uniq_name = "_QFEunit"}
+  integer :: unit
+  character(7) :: c1
+  character(4) :: c2
+  integer :: size = 0
+  integer :: stat = 0
+  ! CHECK: %[[cookie:.*]] = fir.call @_FortranAioBeginOpenNewUnit(%{{.*}}, %{{.*}}) : (!fir.ref<i8>, i32) -> !fir.ref<i8>
+  ! CHECK: fir.call @_FortranAioSetAccess(%[[cookie]], %{{.*}}, %{{.*}}) : (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
+  ! CHECK: fir.call @_FortranAioSetAction(%[[cookie]], %{{.*}}, %{{.*}}) : (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
+  ! CHECK: fir.call @_FortranAioSetForm(%[[cookie]], %{{.*}}, %{{.*}}) : (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
+  ! CHECK: fir.call @_FortranAioSetStatus(%[[cookie]], %{{.*}}, %{{.*}}) : (!fir.ref<i8>, !fir.ref<i8>, i64) -> i1
+  ! CHECK: %[[kind:.*]] = arith.constant 4 : i32
+  ! CHECK: fir.call @_FortranAioGetNewUnit(%[[cookie]], %[[unit]], %[[kind]]) : (!fir.ref<i8>, !fir.ref<i32>, i32) -> i1
+  ! CHECK: fir.call @_FortranAioEndIoStatement(%[[cookie]]) : (!fir.ref<i8>) -> i32
+  OPEN(NEWUNIT=unit,ACCESS='SEQUENTIAL',ACTION='READWRITE',&
+    FORM='FORMATTED',STATUS='SCRATCH')
+  WRITE(unit, '(A)') "ABCDEF"
+  WRITE(unit, '(A)') "GHIJKL"
+  REWIND(unit)
+  call test_read_size(size, c1, c2, unit, stat)
+  print *, stat, size, c1
+  CLOSE(unit)
+end