[Mlir-commits] [mlir] 469096d - [mlir][Linalg] Fix tests in tile-and-pad
Hanhan Wang
llvmlistbot at llvm.org
Thu Jan 28 08:05:37 PST 2021
Author: Hanhan Wang
Date: 2021-01-28T07:59:33-08:00
New Revision: 469096d18efde7622b2bc41d313e58ab7789d40e
URL: https://github.com/llvm/llvm-project/commit/469096d18efde7622b2bc41d313e58ab7789d40e
DIFF: https://github.com/llvm/llvm-project/commit/469096d18efde7622b2bc41d313e58ab7789d40e.diff
LOG: [mlir][Linalg] Fix tests in tile-and-pad
The check match in D95555 was wrong, this patch fixes it.
Differential Revision: https://reviews.llvm.org/D95618
Added:
Modified:
mlir/test/Dialect/Linalg/tile-and-pad-tensors.mlir
Removed:
################################################################################
diff --git a/mlir/test/Dialect/Linalg/tile-and-pad-tensors.mlir b/mlir/test/Dialect/Linalg/tile-and-pad-tensors.mlir
index 4eaf8300269a..98c3511590bf 100644
--- a/mlir/test/Dialect/Linalg/tile-and-pad-tensors.mlir
+++ b/mlir/test/Dialect/Linalg/tile-and-pad-tensors.mlir
@@ -18,12 +18,12 @@ func @matmul_tensors(
// CHECK-NOT: linalg.matmul {{.*}} tensor<?x?xf32>
// Padding injects static information.
-// CHECK: %[[pA:.*]] = linalg.pad_tensor %[[sTA]] low[0, 0] high[%{{.*}}, %{{.*}}]
-// CHECK: : tensor<?x?xf32> to tensor<2x4xf32>
-// CHECK: %[[pB:.*]] = linalg.pad_tensor %[[sTB]] low[0, 0] high[%{{.*}}, %{{.*}}]
-// CHECK: : tensor<?x?xf32> to tensor<2x4xf32>
-// CHECK: %[[pC:.*]] = linalg.pad_tensor %[[sTC]] low[0, 0] high[%{{.*}}, %{{.*}}]
+// CHECK: %[[pA:.*]] = linalg.pad_tensor %[[sTA]] low[%c0, %c0] high[%{{.*}}, %{{.*}}]
// CHECK: : tensor<?x?xf32> to tensor<2x4xf32>
+// CHECK: %[[pB:.*]] = linalg.pad_tensor %[[sTB]] low[%c0, %c0] high[%{{.*}}, %{{.*}}]
+// CHECK: : tensor<?x?xf32> to tensor<4x3xf32>
+// CHECK: %[[pC:.*]] = linalg.pad_tensor %[[sTC]] low[%c0, %c0] high[%{{.*}}, %{{.*}}]
+// CHECK: : tensor<?x?xf32> to tensor<2x3xf32>
// CHECK: %[[pD:.*]] = linalg.matmul ins(%[[pA]], %[[pB]] : tensor<2x4xf32>, tensor<4x3xf32>)
// CHECK-SAME: outs(%[[pC]] : tensor<2x3xf32>) -> tensor<2x3xf32>
// CHECK: %[[sTD:.*]] = subtensor %[[pD]][0, 0] [%{{.*}}, %{{.*}}] [1, 1] : tensor<2x3xf32> to tensor<?x?xf32>
More information about the Mlir-commits
mailing list