Skip to content

Commit 0cf035c

Browse files
committed
modify test
1 parent 4dde55a commit 0cf035c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

mlir/test/Dialect/Vector/vector-unroll-options.mlir

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -222,13 +222,15 @@ func.func @vector_multi_reduction(%v : vector<4x6xf32>, %acc: vector<4xf32>) ->
222222
// CHECK: %[[V2:.*]] = vector.insert_strided_slice %[[R5]], %[[V1]] {offsets = [2], strides = [1]} : vector<2xf32> into vector<4xf32>
223223
// CHECK: return %[[V2]] : vector<4xf32>
224224

225+
// This is a negative test case to ensure that further unrolling is not performed. Since the vector.multi_reduction
226+
// operation has already been unrolled, attempting additional unrolling should not be allowed.
225227
func.func @negative_vector_multi_reduction(%v: vector<4x2xf32>, %acc: f32) -> f32 {
226228
%0 = vector.multi_reduction #vector.kind<add>, %v, %acc [0, 1] : vector<4x2xf32> to f32
227229
return %0 : f32
228230
}
229231
// CHECK-LABEL: func @negative_vector_multi_reduction
230-
// CHECK: %[[R0:.*]] = vector.multi_reduction <add>, %{{.*}}, %{{.*}} [0, 1] : vector<4x2xf32> to f32
231-
// CHECK: return %[[R0]] : f32
232+
// CHECK-NEXT: %[[R0:.*]] = vector.multi_reduction <add>, %{{.*}}, %{{.*}} [0, 1] : vector<4x2xf32> to f32
233+
// CHECK-NEXT: return %[[R0]] : f32
232234

233235
func.func @vector_reduction(%v : vector<8xf32>) -> f32 {
234236
%0 = vector.reduction <add>, %v : vector<8xf32> into f32

0 commit comments

Comments
 (0)