Skip to content

Commit 74ec44a

Browse files
committed
modify test
1 parent 04d4303 commit 74ec44a

File tree

1 file changed

+6
-11
lines changed

1 file changed

+6
-11
lines changed

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

Lines changed: 6 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -188,18 +188,13 @@ func.func @vector_fma(%a: vector<4x4xf32>, %b: vector<4x4xf32>, %c: vector<4x4xf
188188
// CHECK-LABEL: func @vector_fma
189189
// CHECK-COUNT-4: vector.fma %{{.+}}, %{{.+}}, %{{.+}} : vector<2x2xf32>
190190

191-
func.func @higher_rank_unroll() {
192-
%cst_25 = arith.constant dense<3.718400e+04> : vector<4x2x2xf16>
193-
%cst_26 = arith.constant dense<1.000000e+00> : vector<24x2x2xf32>
194-
%47 = vector.fma %cst_26, %cst_26, %cst_26 : vector<24x2x2xf32>
195-
%818 = scf.execute_region -> vector<24x2x2xf32> {
196-
scf.yield %47 : vector<24x2x2xf32>
197-
}
198-
%823 = vector.extract_strided_slice %cst_25 {offsets = [2], sizes = [1], strides = [1]} : vector<4x2x2xf16> to vector<1x2x2xf16>
199-
return
191+
func.func @negative_vector_fma_3d(%a: vector<3x2x2xf32>) {
192+
%0 = vector.fma %a, %a, %a : vector<3x2x2xf32>
193+
return
200194
}
201-
202-
// CHECK-LABEL: func @higher_rank_unroll
195+
// CHECK-LABEL: func @negative_vector_fma_3d
196+
// CHECK-NOT: vector.extract_strided_slice
197+
// CHECK: %0 = vector.fma %a, %a, %a : vector<3x2x2xf32>
203198
// CHECK: return
204199

205200
func.func @vector_multi_reduction(%v : vector<4x6xf32>, %acc: vector<4xf32>) -> vector<4xf32> {

0 commit comments

Comments
 (0)