Skip to content

Commit 822969d

Browse files
committed
fixup! fixup! [mlir][linalg] Add e2e test for linalg.mmt4d
Address PR comments
1 parent d505a21 commit 822969d

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

mlir/test/Integration/Dialect/Linalg/CPU/pack-unpack-mmt4d.mlir

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,12 @@
11
// DEFINE: %{compile} = mlir-opt %s \
22
// DEFINE: -transform-interpreter -test-transform-dialect-erase-schedule \
3-
// DEFINE: -one-shot-bufferize="bufferize-function-boundaries" -test-lower-to-llvm -o %t -o %t
3+
// DEFINE: -one-shot-bufferize="bufferize-function-boundaries" \
4+
// DEFINE: -buffer-deallocation-pipeline -cse -canonicalize -convert-vector-to-scf -test-lower-to-llvm
45
// DEFINE: %{entry_point} = main
5-
// DEFINE: %{run} = mlir-cpu-runner %t -e %{entry_point} -entry-point-result=void \
6+
// DEFINE: %{run} = mlir-cpu-runner -e %{entry_point} -entry-point-result=void \
67
// DEFINE: -shared-libs=%mlir_runner_utils,%mlir_c_runner_utils
78

8-
// RUN: %{compile}
9-
10-
// RUN: %{run} | FileCheck %s
9+
// RUN: %{compile} | %{run} | FileCheck %s
1110

1211
/// End-to-end test for computing matrix-multiplication using linalg.mmt4d. In
1312
/// particular, demonstrates how the following MLIR sequence (implemented in @mmt4d):
@@ -81,7 +80,6 @@ func.func @matmul(%A: tensor<7x16xi32>, %B: tensor<16x13xi32>, %C: tensor<7x13xi
8180
func.func @mmt4d(%A: tensor<7x16xi32>, %B: tensor<16x13xi32>, %C: tensor<7x13xi32>) -> tensor<7x13xi32> {
8281
%zero = arith.constant 0 : i32
8382

84-
%cst = arith.constant 0 : i32
8583
%A_pack_empty = tensor.empty() : tensor<2x16x8x1xi32>
8684
%B_pack_empty = tensor.empty() : tensor<2x16x8x1xi32>
8785
%C_pack_empty = tensor.empty() : tensor<2x2x8x8xi32>

0 commit comments

Comments
 (0)