1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// RUN: mlir-opt %s -test-wrap-scf-while-loop-in-zero-trip-check -split-input-file | FileCheck %s
// RUN: mlir-opt %s -test-wrap-scf-while-loop-in-zero-trip-check='force-create-check=true' -split-input-file | FileCheck %s --check-prefix FORCE-CREATE-CHECK
func.func @wrap_while_loop_in_zero_trip_check(%bound : i32) -> i32 {
%cst0 = arith.constant 0 : i32
%cst5 = arith.constant 5 : i32
%res:2 = scf.while (%iter = %cst0) : (i32) -> (i32, i32) {
%cond = arith.cmpi slt, %iter, %bound : i32
%inv = arith.addi %bound, %cst5 : i32
scf.condition(%cond) %iter, %inv : i32, i32
} do {
^bb0(%arg1: i32, %arg2: i32):
%next = arith.addi %arg1, %arg2 : i32
scf.yield %next : i32
}
return %res#0 : i32
}
// CHECK-LABEL: func.func @wrap_while_loop_in_zero_trip_check(
// CHECK-SAME: %[[BOUND:.*]]: i32) -> i32 {
// CHECK-DAG: %[[C0:.*]] = arith.constant 0 : i32
// CHECK-DAG: %[[C5:.*]] = arith.constant 5 : i32
// CHECK-DAG: %[[PRE_COND:.*]] = arith.cmpi slt, %[[C0]], %[[BOUND]] : i32
// CHECK-DAG: %[[PRE_INV:.*]] = arith.addi %[[BOUND]], %[[C5]] : i32
// CHECK: %[[IF:.*]]:2 = scf.if %[[PRE_COND]] -> (i32, i32) {
// CHECK: %[[WHILE:.*]]:2 = scf.while (
// CHECK-SAME: %[[ARG1:.*]] = %[[C0]], %[[ARG2:.*]] = %[[PRE_INV]]
// CHECK-SAME: ) : (i32, i32) -> (i32, i32) {
// CHECK: %[[NEXT:.*]] = arith.addi %[[ARG1]], %[[ARG2]] : i32
// CHECK: %[[COND:.*]] = arith.cmpi slt, %[[NEXT]], %[[BOUND]] : i32
// CHECK: %[[INV:.*]] = arith.addi %[[BOUND]], %[[C5]] : i32
// CHECK: scf.condition(%[[COND]]) %[[NEXT]], %[[INV]] : i32, i32
// CHECK: } do {
// CHECK: ^bb0(%[[ARG3:.*]]: i32, %[[ARG4:.*]]: i32):
// CHECK: scf.yield %[[ARG3]], %[[ARG4]] : i32, i32
// CHECK: }
// CHECK: scf.yield %[[WHILE]]#0, %[[WHILE]]#1 : i32, i32
// CHECK: } else {
// CHECK: scf.yield %[[C0]], %[[PRE_INV]] : i32, i32
// CHECK: }
// CHECK: return %[[IF]]#0 : i32
// -----
func.func @wrap_while_loop_with_minimal_before_block(%bound : i32) -> i32 {
%cst0 = arith.constant 0 : i32
%true = arith.constant true
%cst5 = arith.constant 5 : i32
%res = scf.while (%iter = %cst0, %arg0 = %true) : (i32, i1) -> i32 {
scf.condition(%arg0) %iter : i32
} do {
^bb0(%arg1: i32):
%next = arith.addi %arg1, %cst5 : i32
%cond = arith.cmpi slt, %next, %bound : i32
scf.yield %next, %cond : i32, i1
}
return %res : i32
}
// CHECK-LABEL: func.func @wrap_while_loop_with_minimal_before_block(
// CHECK-SAME: %[[BOUND:.*]]: i32) -> i32 {
// CHECK-DAG: %[[C0:.*]] = arith.constant 0 : i32
// CHECK-DAG: %[[TRUE:.*]] = arith.constant true
// CHECK-DAG: %[[C5:.*]] = arith.constant 5 : i32
// CHECK: %[[IF:.*]] = scf.if %[[TRUE]] -> (i32) {
// CHECK: %[[WHILE:.*]] = scf.while (%[[ARG1:.*]] = %[[C0]]) : (i32) -> i32 {
// CHECK: %[[NEXT:.*]] = arith.addi %[[ARG1]], %[[C5]] : i32
// CHECK: %[[COND:.*]] = arith.cmpi slt, %[[NEXT]], %[[BOUND]] : i32
// CHECK: scf.condition(%[[COND]]) %[[NEXT]] : i32
// CHECK: } do {
// CHECK: ^bb0(%[[ARG2:.*]]: i32):
// CHECK: scf.yield %[[ARG2]] : i32
// CHECK: }
// CHECK: scf.yield %[[WHILE]] : i32
// CHECK: } else {
// CHECK: scf.yield %[[C0]] : i32
// CHECK: }
// CHECK: return %[[IF]] : i32
// -----
func.func @wrap_do_while_loop_in_zero_trip_check(%bound : i32) -> i32 {
%cst0 = arith.constant 0 : i32
%cst5 = arith.constant 5 : i32
%res = scf.while (%iter = %cst0) : (i32) -> i32 {
%next = arith.addi %iter, %cst5 : i32
%cond = arith.cmpi slt, %next, %bound : i32
scf.condition(%cond) %next : i32
} do {
^bb0(%arg1: i32):
scf.yield %arg1 : i32
}
return %res : i32
}
// CHECK-LABEL: func.func @wrap_do_while_loop_in_zero_trip_check(
// CHECK-SAME: %[[BOUND:.*]]: i32) -> i32 {
// CHECK-DAG: %[[C0:.*]] = arith.constant 0 : i32
// CHECK-DAG: %[[C5:.*]] = arith.constant 5 : i32
// CHECK-NOT: scf.if
// CHECK: %[[WHILE:.*]] = scf.while (%[[ARG1:.*]] = %[[C0]]) : (i32) -> i32 {
// CHECK: %[[NEXT:.*]] = arith.addi %[[ARG1]], %[[C5]] : i32
// CHECK: %[[COND:.*]] = arith.cmpi slt, %[[NEXT]], %[[BOUND]] : i32
// CHECK: scf.condition(%[[COND]]) %[[NEXT]] : i32
// CHECK: } do {
// CHECK: ^bb0(%[[ARG2:.*]]: i32):
// CHECK: scf.yield %[[ARG2]] : i32
// CHECK: }
// CHECK: return %[[WHILE]] : i32
// FORCE-CREATE-CHECK-LABEL: func.func @wrap_do_while_loop_in_zero_trip_check(
// FORCE-CREATE-CHECK-SAME: %[[BOUND:.*]]: i32) -> i32 {
// FORCE-CREATE-CHECK-DAG: %[[C0:.*]] = arith.constant 0 : i32
// FORCE-CREATE-CHECK-DAG: %[[C5:.*]] = arith.constant 5 : i32
// FORCE-CREATE-CHECK: %[[PRE_NEXT:.*]] = arith.addi %[[C0]], %[[C5]] : i32
// FORCE-CREATE-CHECK: %[[PRE_COND:.*]] = arith.cmpi slt, %[[PRE_NEXT]], %[[BOUND]] : i32
// FORCE-CREATE-CHECK: %[[IF:.*]] = scf.if %[[PRE_COND]] -> (i32) {
// FORCE-CREATE-CHECK: %[[WHILE:.*]] = scf.while (%[[ARG1:.*]] = %[[PRE_NEXT]]) : (i32) -> i32 {
// FORCE-CREATE-CHECK: %[[NEXT:.*]] = arith.addi %[[ARG1]], %[[C5]] : i32
// FORCE-CREATE-CHECK: %[[COND:.*]] = arith.cmpi slt, %[[NEXT]], %[[BOUND]] : i32
// FORCE-CREATE-CHECK: scf.condition(%[[COND]]) %[[NEXT]] : i32
// FORCE-CREATE-CHECK: } do {
// FORCE-CREATE-CHECK: ^bb0(%[[ARG2:.*]]: i32):
// FORCE-CREATE-CHECK: scf.yield %[[ARG2]] : i32
// FORCE-CREATE-CHECK: }
// FORCE-CREATE-CHECK: scf.yield %[[WHILE]] : i32
// FORCE-CREATE-CHECK: } else {
// FORCE-CREATE-CHECK: scf.yield %[[PRE_NEXT]] : i32
// FORCE-CREATE-CHECK: }
// FORCE-CREATE-CHECK: return %[[IF]] : i32
|
