Skip to content

Commit 4154cb1

Browse files
author
ZhiQiang Fan
committed
[flang][fir] Update test file and formet data type
1 parent d6bc837 commit 4154cb1

File tree

2 files changed

+33
-73
lines changed

2 files changed

+33
-73
lines changed

‎flang/lib/Optimizer/Transforms/FIRToSCF.cpp

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -96,14 +96,14 @@ struct IfConversion : public OpRewritePattern<fir::IfOp> {
9696
mlir::detail::TypedValue<mlir::IntegerType> condition = ifOp.getCondition();
9797
ValueTypeRange<ResultRange> resultTypes = ifOp.getResultTypes();
9898
bool hasResult = !resultTypes.empty();
99-
auto scfIfOp = rewriter.create<scf::IfOp>(loc, resultTypes, condition,
100-
!ifOp.getElseRegion().empty());
99+
mlir::scf::IfOp scfIfOp = rewriter.create<scf::IfOp>(
100+
loc, resultTypes, condition, !ifOp.getElseRegion().empty());
101101
// then region
102102
assert(!ifOp.getThenRegion().empty() && "must have then region");
103-
auto &firThenBlock = ifOp.getThenRegion().front();
104-
auto &scfThenBlock = scfIfOp.getThenRegion().front();
105-
auto &firThenOps = firThenBlock.getOperations();
106-
mlir::Operation *firThenTerminator = firThenBlock.getTerminator();
103+
Block &firThenBlock = ifOp.getThenRegion().front();
104+
Block &scfThenBlock = scfIfOp.getThenRegion().front();
105+
llvm::iplist<Operation> &firThenOps = firThenBlock.getOperations();
106+
Operation *firThenTerminator = firThenBlock.getTerminator();
107107

108108
rewriter.setInsertionPointToStart(&scfThenBlock);
109109
// not splice terminator
@@ -119,9 +119,9 @@ struct IfConversion : public OpRewritePattern<fir::IfOp> {
119119

120120
// else region
121121
if (!ifOp.getElseRegion().empty()) {
122-
auto &firElseBlock = ifOp.getElseRegion().front();
123-
auto &scfElseBlock = scfIfOp.getElseRegion().front();
124-
auto &firElseOps = firElseBlock.getOperations();
122+
Block &firElseBlock = ifOp.getElseRegion().front();
123+
Block &scfElseBlock = scfIfOp.getElseRegion().front();
124+
llvm::iplist<Operation> &firElseOps = firElseBlock.getOperations();
125125
mlir::Operation *firElseTerminator = firElseBlock.getTerminator();
126126

127127
rewriter.setInsertionPointToStart(&scfElseBlock);

‎flang/test/Fir/FirToSCF/if.fir

Lines changed: 24 additions & 64 deletions
Original file line numberDiff line numberDiff line change
@@ -1,97 +1,57 @@
1-
21
// RUN: fir-opt %s --fir-to-scf | FileCheck %s
32

4-
// CHECK: func.func @_QFPtest_only(
5-
// CHECK: %[[ARG0:.*]]: !fir.ref<tuple<!fir.ref<f32>>>) {
6-
// CHECK: %[[VAL_0:.*]] = arith.constant 1 : i32
7-
// CHECK: %[[VAL_1:.*]] = arith.constant false
8-
// CHECK: %[[VAL_2:.*]] = arith.constant 0 : i32
9-
// CHECK: scf.if %[[VAL_1:.*]] {
10-
// CHECK: %[[VAL_3:.*]] = arith.addi %[[VAL_2:.*]], %[[VAL_0:.*]] : i32
3+
// CHECK: func.func @test_only(%[[ARG0:.*]]: i1, %[[ARG1:.*]]: i32) {
4+
// CHECK: scf.if %[[ARG0:.*]] {
5+
// CHECK: %[[VAL_1:.*]] = arith.addi %[[ARG1:.*]], %[[ARG1:.*]] : i32
116
// CHECK: }
127
// CHECK: return
138
// CHECK: }
14-
func.func @_QFPtest_only(%arg0: !fir.ref<tuple<!fir.ref<f32>>>) {
15-
%c1_i32 = arith.constant 1 : i32
16-
%false = arith.constant false
17-
%c0_i32 = arith.constant 0 : i32
18-
fir.if %false {
19-
%0 = arith.addi %c0_i32, %c1_i32 : i32
9+
func.func @test_only(%arg0 : i1, %arg1 : i32) {
10+
fir.if %arg0 {
11+
%0 = arith.addi %arg1, %arg1 : i32
2012
}
2113
return
2214
}
2315

24-
// CHECK: func.func @_QFPtest_else(
25-
// CHECK: %[[ARG0:.*]]: !fir.ref<tuple<!fir.ref<f32>>>) {
26-
// CHECK: %[[VAL_0:.*]] = arith.constant 1 : i32
16+
// CHECK: func.func @test_else() {
2717
// CHECK: %[[VAL_1:.*]] = arith.constant false
28-
// CHECK: %[[VAL_2:.*]] = arith.constant 0 : i32
29-
// CHECK: %[[VAL_3:.*]] = fir.dummy_scope : !fir.dscope
30-
// CHECK: %[[VAL_4:.*]] = fir.coordinate_of %[[ARG0:.*]], %[[VAL_2:.*]] : (!fir.ref<tuple<!fir.ref<f32>>>, i32) -> !fir.llvm_ptr<!fir.ref<f32>>
31-
// CHECK: %[[VAL_5:.*]] = fir.load %[[VAL_4:.*]] : !fir.llvm_ptr<!fir.ref<f32>>
32-
// CHECK: %[[VAL_6:.*]] = fir.declare %[[VAL_5:.*]] {fortran_attrs = #fir.var_attrs<host_assoc>, uniq_name = "_QFEx"} : (!fir.ref<f32>) -> !fir.ref<f32>
33-
// CHECK: %[[VAL_7:.*]] = fir.address_of(@_QFFtest_elseEsum) : !fir.ref<i32>
34-
// CHECK: %[[VAL_10:.*]] = fir.declare %[[VAL_7:.*]] {uniq_name = "_QFFtest_elseEsum"} : (!fir.ref<i32>) -> !fir.ref<i32>
18+
// CHECK: %[[VAL_2:.*]] = arith.constant 2 : i32
3519
// CHECK: scf.if %[[VAL_1:.*]] {
36-
// CHECK: %[[VAL_8:.*]] = fir.load %[[VAL_10:.*]] : !fir.ref<i32>
37-
// CHECK: %[[VAL_9:.*]] = arith.addi %[[VAL_8:.*]], %[[VAL_0:.*]] : i32
38-
// CHECK: fir.store %[[VAL_9:.*]] to %[[VAL_10:.*]] : !fir.ref<i32>
20+
// CHECK: %[[VAL_3:.*]] = arith.constant 3 : i32
3921
// CHECK: } else {
40-
// CHECK: %[[VAL_8:.*]] = fir.load %[[VAL_10:.*]] : !fir.ref<i32>
41-
// CHECK: %[[VAL_9:.*]] = arith.addi %[[VAL_8:.*]], %c1_i32 : i32
42-
// CHECK: fir.store %[[VAL_9:.*]] to %[[VAL_10:.*]] : !fir.ref<i32>
22+
// CHECK: %[[VAL_3:.*]] = arith.constant 3 : i32
4323
// CHECK: }
4424
// CHECK: return
4525
// CHECK: }
46-
func.func @_QFPtest_else(%arg0: !fir.ref<tuple<!fir.ref<f32>>> {}) attributes {} {
47-
%c1_i32 = arith.constant 1 : i32
26+
func.func @test_else() {
4827
%false = arith.constant false
49-
%c0_i32 = arith.constant 0 : i32
50-
%0 = fir.dummy_scope : !fir.dscope
51-
%1 = fir.coordinate_of %arg0, %c0_i32 : (!fir.ref<tuple<!fir.ref<f32>>>, i32) -> !fir.llvm_ptr<!fir.ref<f32>>
52-
%2 = fir.load %1 : !fir.llvm_ptr<!fir.ref<f32>>
53-
%3 = fir.declare %2 {fortran_attrs = #fir.var_attrs<host_assoc>, uniq_name = "_QFEx"} : (!fir.ref<f32>) -> !fir.ref<f32>
54-
%4 = fir.address_of(@_QFFtest_elseEsum) : !fir.ref<i32>
55-
%5 = fir.declare %4 {uniq_name = "_QFFtest_elseEsum"} : (!fir.ref<i32>) -> !fir.ref<i32>
28+
%1 = arith.constant 2 : i32
5629
fir.if %false {
57-
%6 = fir.load %5 : !fir.ref<i32>
58-
%7 = arith.addi %6, %c1_i32 : i32
59-
fir.store %7 to %5 : !fir.ref<i32>
30+
%2 = arith.constant 3 : i32
6031
} else {
61-
%6 = fir.load %5 : !fir.ref<i32>
62-
%7 = arith.addi %6, %c1_i32 : i32
63-
fir.store %7 to %5 : !fir.ref<i32>
32+
%3 = arith.constant 3 : i32
6433
}
6534
return
6635
}
6736

68-
// CHECK-LABEL: func.func @test_two_result() {
69-
// CHECK: %[[VAL_0:.*]] = arith.constant 10 : i32
70-
// CHECK: %[[VAL_1:.*]] = arith.constant 5 : i32
71-
// CHECK: %[[VAL_2:.*]] = arith.cmpi sgt, %[[VAL_0:.*]], %[[VAL_1:.*]] : i32
72-
// CHECK: %[[VAL_3:.*]] = arith.constant 3.140000e+00 : f32
73-
// CHECK: %[[VAL_4:.*]] = arith.constant 2.710000e+00 : f32
74-
// CHECK: %[[VAL_5:.*]] = arith.constant 1.000000e+00 : f32
75-
// CHECK: %[[VAL_6:.*]] = arith.constant 2.000000e+00 : f32
37+
// CHECK-LABEL: func.func @test_two_result() {
38+
// CHECK: %[[VAL_1:.*]] = arith.constant 2.000000e+00 : f32
39+
// CHECK: %[[VAL_2:.*]] = arith.constant false
7640
// CHECK: %[[RES:[0-9]+]]:2 = scf.if %[[VAL_2:.*]] -> (f32, f32) {
77-
// CHECK: scf.yield %[[VAL_3:.*]], %[[VAL_4:.*]] : f32, f32
41+
// CHECK: scf.yield %[[VAL_1:.*]], %[[VAL_1:.*]] : f32, f32
7842
// CHECK: } else {
79-
// CHECK: scf.yield %[[VAL_5:.*]], %[[VAL_6:.*]] : f32, f32
43+
// CHECK: scf.yield %[[VAL_1:.*]], %[[VAL_1:.*]] : f32, f32
8044
// CHECK: }
8145
// CHECK: return
8246
// CHECK: }
47+
8348
func.func @test_two_result() {
84-
%c10_i32 = arith.constant 10 : i32
85-
%c5_i32 = arith.constant 5 : i32
86-
%cmp = arith.cmpi sgt, %c10_i32, %c5_i32 : i32
87-
%c3_14_f32 = arith.constant 3.14 : f32
88-
%c2_71_f32 = arith.constant 2.71 : f32
89-
%c1_0_f32 = arith.constant 1.0 : f32
90-
%c2_0_f32 = arith.constant 2.0 : f32
49+
%1 = arith.constant 2.0 : f32
50+
%cmp = arith.constant false
9151
%x, %y = fir.if %cmp -> (f32, f32) {
92-
fir.result %c3_14_f32, %c2_71_f32 : f32, f32
52+
fir.result %1, %1 : f32, f32
9353
} else {
94-
fir.result %c1_0_f32, %c2_0_f32 : f32, f32
54+
fir.result %1, %1 : f32, f32
9555
}
9656
return
9757
}

0 commit comments

Comments
 (0)