maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[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>
This commit is contained in:
Valentin Clement 2022-03-17 16:59:31 +01:00
parent b09439e20b
commit d670e10235
No known key found for this signature in database
GPG Key ID: 086D54783C928776
1 changed files with 49 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
flang/test/Lower/io-statement-3.f90 Normal file
View File

@ -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