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
131
132
133
134
135
136
137
138
139
140
141
142
143
|
// RUN: mlir-opt %s --mlir-disable-threading -test-block-is-in-loop 2>&1 | FileCheck %s
module {
// Test function with only one bb
func.func @simple() {
func.return
}
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb0:
// Test simple loop bb0 -> bb0
func.func @loopForever() {
^bb0:
cf.br ^bb1
^bb1:
cf.br ^bb1
}
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb0:
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb1:
// Test bb0 -> bb1 -> bb2 -> bb1
func.func @loopForever2() {
^bb0:
cf.br ^bb1
^bb1:
cf.br ^bb2
^bb2:
cf.br ^bb1
}
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb0:
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb1:
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb2:
// Test conditional branch without loop
// bb0 -> bb1 -> {bb2, bb3}
func.func @noLoop(%arg0: i1) {
cf.br ^bb1
^bb1:
cf.cond_br %arg0, ^bb2, ^bb3
^bb2:
func.return
^bb3:
func.return
}
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb0(%arg0: i1)
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb1:
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb2:
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb3:
// test multiple loops
// bb0 -> bb1 -> bb2 -> bb3 { -> bb2} -> bb4 { -> bb1 } -> bb5
func.func @multipleLoops(%arg0: i1, %arg1: i1) {
cf.br ^bb1
^bb1:
cf.br ^bb2
^bb2:
cf.br ^bb3
^bb3:
cf.cond_br %arg0, ^bb4, ^bb2
^bb4:
cf.cond_br %arg1, ^bb1, ^bb5
^bb5:
return
}
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb0(%arg0: i1, %arg1: i1)
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb1:
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb2:
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb3:
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb4:
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb5:
// test derived from real Flang output
func.func @_QPblockTest0(%arg0: i1, %arg1: i1) {
cf.br ^bb1
^bb1: // 2 preds: ^bb0, ^bb4
cf.cond_br %arg0, ^bb2, ^bb5
^bb2: // pred: ^bb1
cf.cond_br %arg1, ^bb3, ^bb4
^bb3: // pred: ^bb2
return
^bb4: // pred: ^bb2
cf.br ^bb1
^bb5: // pred: ^bb1
return
}
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb0(%arg0: i1, %arg1: i1)
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb1:
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb2:
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb3:
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb4:
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb5:
// check nested blocks
func.func @check_alloc_in_loop(%counter : i64) {
cf.br ^bb1(%counter: i64)
^bb1(%lv : i64):
%cm1 = arith.constant -1 : i64
%rem = arith.addi %lv, %cm1 : i64
%zero = arith.constant 0 : i64
%p = arith.cmpi eq, %rem, %zero : i64
cf.cond_br %p, ^bb3, ^bb2
^bb2:
scf.execute_region -> () {
%c1 = arith.constant 1 : i64
scf.yield
}
cf.br ^bb1(%rem: i64)
^bb3:
return
}
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb0(%arg0: i64):
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb1(%0: i64)
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb0:
// CHECK-NEXT: %c1_i64
// CHECK: Block is in a loop
// CHECK-NEXT: ^bb2:
// CHECK: Block is not in a loop
// CHECK-NEXT: ^bb3:
}
|
