Skip to content

Commit bd61b2c

Browse files
committed
fixup! [mlir][SVE] Add an e2e test for vectorization of linalg.matmul
Simplity check-lines
1 parent f39c97d commit bd61b2c

File tree

1 file changed

+10
-6
lines changed
  • mlir/test/Integration/Dialect/Linalg/CPU/ArmSVE

1 file changed

+10
-6
lines changed

mlir/test/Integration/Dialect/Linalg/CPU/ArmSVE/matmul.mlir

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -25,17 +25,21 @@ func.func @entry() {
2525

2626
%C_out = linalg.matmul ins(%A, %B: tensor<?x?xf32>, tensor<?x?xf32>) outs(%C_in: tensor<?x?xf32>) -> tensor<?x?xf32>
2727

28-
// There are at least 4 f32 elements in every SVE vector, i.e.
29-
// * %vscale is >= 1.
30-
// For implementations with wider vectors, you should see more elements being
31-
// printed.
32-
// CHECK: {{\[}}[9.8596, 9.8596, 9.8596, 9.8596
28+
// CHECK-LABEL: SVE: START OF TEST OUTPUT
29+
vector.print str "SVE: START OF TEST OUTPUT"
30+
31+
// There are at least 4 x f32 elements in every SVE vector, i.e.
32+
// * %vscale >= 1.
33+
// Hence, when checking the outupt there will always be at least 4 elements
34+
// in every row. For implementations with wider vectors, you should see more
35+
// elements being printed.
36+
// CHECK: [9.8596, 9.8596, 9.8596, 9.8596
3337
// CHECK-NEXT: [9.8596, 9.8596, 9.8596, 9.8596
3438

3539
%xf = tensor.cast %C_out : tensor<?x?xf32> to tensor<*xf32>
3640
call @printMemrefF32(%xf) : (tensor<*xf32>) -> ()
3741

38-
// CHECK: SVE: END OF TEST OUTPUT
42+
// CHECK-NEXT: SVE: END OF TEST OUTPUT
3943
vector.print str "SVE: END OF TEST OUTPUT"
4044

4145
return

0 commit comments

Comments
 (0)