[Mlir-commits] [mlir] [mlir][Vector] Add missing CHECK rules to vector-transfer-flatten.mlir (PR #82698)

[Diego Caballero]

https://github.com/dcaballe created https://github.com/llvm/llvm-project/pull/82698

This test failed after landing #81964 due to a bad merge. I provided a quick fix and this PR is adding the rest of CHECK rules that were not merged properly.

>From e754ccb91798565b54189f756e4a81b403f72698 Mon Sep 17 00:00:00 2001
From: Diego Caballero <diegocaballero at google.com>
Date: Thu, 22 Feb 2024 22:11:47 +0000
Subject: [PATCH] [mlir][Vector] Add missing CHECK rules to
 vector-transfer-flatten.mlir

This test failed after landing #81964 due to a bad merge. I provided a
quick fix and this PR is adding the rest of CHECK rules that were not
merged properly.
---
 mlir/test/Dialect/Vector/vector-transfer-flatten.mlir | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mlir/test/Dialect/Vector/vector-transfer-flatten.mlir b/mlir/test/Dialect/Vector/vector-transfer-flatten.mlir
index 2766e782a3fb27..788ae9ac044ed8 100644
--- a/mlir/test/Dialect/Vector/vector-transfer-flatten.mlir
+++ b/mlir/test/Dialect/Vector/vector-transfer-flatten.mlir
@@ -476,6 +476,7 @@ func.func @regression_non_contiguous_dim_read(%subview : memref<1x3x3x2xf32, str
 //       CHECK:     %[[APPLY:.*]] = affine.apply #[[$MAP]]()
 
 // CHECK-128B-LABEL: func @regression_non_contiguous_dim_read(
+//       CHECK-128B:   memref.collapse_shape
 
 // -----
 
@@ -491,3 +492,4 @@ func.func @unsupported_non_contiguous_dim_write(%value : vector<2x2xf32>,
 //   CHECK-NOT:    memref.collapse_shape
 
 // CHECK-128B-LABEL: func @unsupported_non_contiguous_dim_write(
+//   CHECK-128B-NOT:   memref.collapse_shape