[flang-commits] [flang] d670e10 - [flang] Add IO lowering test

[Valentin Clement via flang-commits]

Author: Valentin Clement
Date: 2022-03-17T17:00:13+01:00
New Revision: d670e1023550b4f18cbb45116de5418d9cef8cdd

URL: https://github.com/llvm/llvm-project/commit/d670e1023550b4f18cbb45116de5418d9cef8cdd
DIFF: https://github.com/llvm/llvm-project/commit/d670e1023550b4f18cbb45116de5418d9cef8cdd.diff

LOG: [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 at nvidia.com>

Added: 
    flang/test/Lower/io-statement-3.f90

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/flang/test/Lower/io-statement-3.f90 b/flang/test/Lower/io-statement-3.f90
new file mode 100644
index 0000000000000..0d022716e2b75
--- /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