@@ -10,7 +10,7 @@ func @fuse_empty_loops() {
10
10
scf.parallel (%i , %j ) = (%c0 , %c0 ) to (%c2 , %c2 ) step (%c1 , %c1 ) {
11
11
scf.yield
12
12
}
13
- " xla_lhlo.terminator " () : () -> ()
13
+ return
14
14
}
15
15
// CHECK-LABEL: func @fuse_empty_loops
16
16
// CHECK: [[C2:%.*]] = constant 2 : index
@@ -140,7 +140,7 @@ func @do_not_fuse_nested_ploop1() {
140
140
scf.parallel (%i , %j ) = (%c0 , %c0 ) to (%c2 , %c2 ) step (%c1 , %c1 ) {
141
141
scf.yield
142
142
}
143
- " xla_lhlo.terminator " () : () -> ()
143
+ return
144
144
}
145
145
// CHECK-LABEL: func @do_not_fuse_nested_ploop1
146
146
// CHECK: scf.parallel
@@ -162,7 +162,7 @@ func @do_not_fuse_nested_ploop2() {
162
162
}
163
163
scf.yield
164
164
}
165
- " xla_lhlo.terminator " () : () -> ()
165
+ return
166
166
}
167
167
// CHECK-LABEL: func @do_not_fuse_nested_ploop2
168
168
// CHECK: scf.parallel
@@ -181,7 +181,7 @@ func @do_not_fuse_loops_unmatching_num_loops() {
181
181
scf.parallel (%i ) = (%c0 ) to (%c2 ) step (%c1 ) {
182
182
scf.yield
183
183
}
184
- " xla_lhlo.terminator " () : () -> ()
184
+ return
185
185
}
186
186
// CHECK-LABEL: func @do_not_fuse_loops_unmatching_num_loops
187
187
// CHECK: scf.parallel
@@ -200,7 +200,7 @@ func @do_not_fuse_loops_with_side_effecting_ops_in_between() {
200
200
scf.parallel (%i , %j ) = (%c0 , %c0 ) to (%c2 , %c2 ) step (%c1 , %c1 ) {
201
201
scf.yield
202
202
}
203
- " xla_lhlo.terminator " () : () -> ()
203
+ return
204
204
}
205
205
// CHECK-LABEL: func @do_not_fuse_loops_with_side_effecting_ops_in_between
206
206
// CHECK: scf.parallel
@@ -219,7 +219,7 @@ func @do_not_fuse_loops_unmatching_iteration_space() {
219
219
scf.parallel (%i , %j ) = (%c0 , %c0 ) to (%c2 , %c2 ) step (%c1 , %c1 ) {
220
220
scf.yield
221
221
}
222
- " xla_lhlo.terminator " () : () -> ()
222
+ return
223
223
}
224
224
// CHECK-LABEL: func @do_not_fuse_loops_unmatching_iteration_space
225
225
// CHECK: scf.parallel
@@ -302,7 +302,7 @@ func @do_not_fuse_loops_with_memref_defined_in_loop_bodies() {
302
302
%A_elem = load %A [%i , %j ] : memref <?x?xf32 , offset : ?, strides :[?, ?]>
303
303
scf.yield
304
304
}
305
- " xla_lhlo.terminator " () : () -> ()
305
+ return
306
306
}
307
307
// CHECK-LABEL: func @do_not_fuse_loops_with_memref_defined_in_loop_bodies
308
308
// CHECK: scf.parallel
0 commit comments