@@ -1295,24 +1295,6 @@ func.func @no_bubble_up_pack_expanded_padding_through_expand_cannot_reassociate(
1295
1295
1296
1296
// -----
1297
1297
1298
- func.func @bubble_up_pack_extending_dimension_through_expand_can_reassociate (%arg0: tensor <32 x64 xf32 >) -> tensor <4 x4 x16 x8 xf32 > {
1299
- %empty = tensor.empty () : tensor <4 x4 x16 x8 xf32 >
1300
- %expanded = tensor.expand_shape %arg0 [[0 ], [1 , 2 ]] output_shape [32 , 4 , 16 ] : tensor <32 x64 xf32 > into tensor <32 x4 x16 xf32 >
1301
- %pack = linalg.pack %expanded inner_dims_pos = [0 ] inner_tiles = [8 ] into %empty : tensor <32 x4 x16 xf32 > -> tensor <4 x4 x16 x8 xf32 >
1302
- return %pack : tensor <4 x4 x16 x8 xf32 >
1303
- }
1304
- // CHECK-LABEL: func.func @bubble_up_pack_extending_dimension_through_expand_can_reassociate(
1305
- // CHECK-SAME: %[[ARG0:[a-zA-Z0-9]+]]
1306
- // CHECK: %[[EMPTY:.+]] = tensor.empty() : tensor<4x64x8xf32>
1307
- // CHECK: %[[PACK:.+]] = linalg.pack %[[ARG0]]
1308
- // CHECK-SAME: inner_dims_pos = [0] inner_tiles = [8] into %[[EMPTY]]
1309
- // CHECK-SAME: : tensor<32x64xf32> -> tensor<4x64x8xf32>
1310
- // CHECK: %[[EXPANDED:.+]] = tensor.expand_shape %[[PACK]] {{\[}}[0], [1, 2], [3]]
1311
- // CHECK-SAME: output_shape [4, 4, 16, 8] : tensor<4x64x8xf32> into tensor<4x4x16x8xf32>
1312
- // CHECK: return %[[EXPANDED]] : tensor<4x4x16x8xf32>
1313
-
1314
- // -----
1315
-
1316
1298
func.func @push_down_unpack_through_expand (%5: tensor <?x32 x8 x8 xf32 >, %dim: index , %sz0: index ) -> tensor <?x256 x256 xf32 > {
1317
1299
%6 = tensor.empty (%dim ) : tensor <?x256 xf32 >
1318
1300
%unpack = linalg.unpack %5 outer_dims_perm = [0 , 1 ] inner_dims_pos = [0 , 1 ] inner_tiles = [8 , 8 ] into %6 : tensor <?x32 x8 x8 xf32 > -> tensor <?x256 xf32 >
0 commit comments