nit mlir/test/Dialect/Linalg/canonicalize.mlir

Author: ita9naiwa

mlir/test/Dialect/Linalg/canonicalize.mlir

@@ -1760,10 +1760,6 @@ func.func @pack_dont_drop_attributes(%arg0: tensor<?x?x?xf16>, %arg1: tensor<128
 
 // -----
 
-//===----------------------------------------------------------------------===//
-// linalg.unpack + tensor.extract_slice
-
-
 //===----------------------------------------------------------------------===//
 // linalg.fill + linalg.unpack
 //===----------------------------------------------------------------------===//
@@ -1787,6 +1783,30 @@ func.func @fold_dst_style_ops_into_unpack(%arg0 : tensor<?x?x16x64xf32>, %init :
 // tensor.cast + linalg.unpack
 //===----------------------------------------------------------------------===//
 
+// CHECK-LABEL:   func.func @fold_cast_unpack_dynamic_tile_size(
+// CHECK-SAME:      %[[SRC:.*]]: tensor<1x1x8x1xi32>,
+// CHECK-SAME:      %[[DEST:.*]]: tensor<7x?xi32>) -> tensor<7x?xi32> {
+// CHECK:           %[[RES:.*]] = linalg.unpack %[[SRC]] inner_dims_pos = [0, 1] inner_tiles = [8, 1] into %[[DEST]] {test_attr} : tensor<1x1x8x1xi32> -> tensor<7x?xi32>
+// CHECK:           return %[[RES]] : tensor<7x?xi32>
+func.func @fold_cast_unpack_dynamic_tile_size(
+  %src: tensor<1x1x8x1xi32>,
+  %res: tensor<7x?xi32>) -> tensor<7x?xi32> {
+
+    %cast = tensor.cast %src : tensor<1x1x8x1xi32> to tensor<1x1x?x1xi32>
+    %c8 = arith.constant 8 : index
+    %unpack = linalg.unpack %cast
+      inner_dims_pos = [0, 1]
+      inner_tiles = [%c8, 1]
+      into %res {test_attr} : tensor<1x1x?x1xi32> -> tensor<7x?xi32>
+    return %unpack : tensor<7x?xi32>
+}
+
+// -----
+
+//===----------------------------------------------------------------------===//
+// linalg.unpack + tensor.extract_slice
+//===----------------------------------------------------------------------===//
+
 func.func @fold_extract_slice_into_unpack(
     %src : tensor<28x2x?x16x16xf32>, %dest : tensor<28x32x?xf32>, %size : index
 ) -> tensor<28x28x?xf32> {
@@ -1799,6 +1819,7 @@ func.func @fold_extract_slice_into_unpack(
       [0, 0, 0] [28, 28, %size] [1, 1, 1] : tensor<28x32x?xf32> to tensor<28x28x?xf32>
   return %extracted_slice : tensor<28x28x?xf32>
 }
+
 // CHECK-LABEL: func @fold_extract_slice_into_unpack
 //  CHECK-SAME:     %[[SRC:.+]]: tensor<28x2x?x16x16xf32>
 //  CHECK-SAME:     %[[DEST:.+]]: tensor<28x32x?xf32>
@@ -1846,6 +1867,7 @@ func.func @no_fold_extract_slice_into_unpack_non_zero_offset(
       [0, 1] [28, 28] [1, 1] : tensor<28x32xf32> to tensor<28x28xf32>
   return %extracted_slice : tensor<28x28xf32>
 }
+
 // CHECK-LABEL: func @no_fold_extract_slice_into_unpack_non_zero_offset
 //  CHECK-SAME:     %[[SRC:.+]]: tensor<28x2x16xf32>
 //  CHECK-SAME:     %[[DEST:.+]]: tensor<28x32xf32>