Skip to content

Commit 6bc7e37

Browse files
authored
[MLIR][Tensor] Fix checks for fold-into-pack-and-unpack.mlir (#77622)
Fix after 113bce0
1 parent c053e9f commit 6bc7e37

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

mlir/test/Dialect/Tensor/fold-into-pack-and-unpack.mlir

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -469,10 +469,10 @@ func.func @linalg_transpose_tensor_pack_fold_dynamic_outer_dims_tile_dims_tile_s
469469
//CHECK-LABEL: func.func @linalg_transpose_tensor_pack_fold_dynamic_outer_dims_tile_dims_tile_sizes(
470470
// CHECK-SAME: %[[ARG0:.+]]: tensor<?x?x?x?xf32>, %[[ARG1:.+]]: tensor<?x?x?x?xf32>,
471471
// CHECK-SAME: %[[ARG2:.+]]: tensor<?x?x?x?x?x?x?xf32>, %[[ARG3:.+]]: index, %[[ARG4:.+]]: index, %[[ARG5:.+]]: index) -> tensor<?x?x?x?x?x?x?xf32> {
472-
// CHECK: %[[C0:.+]] = arith.constant 0 : index
473-
// CHECK: %[[C1:.+]] = arith.constant 1 : index
474-
// CHECK: %[[C2:.+]] = arith.constant 2 : index
475-
// CHECK: %[[C3:.+]] = arith.constant 3 : index
472+
// CHECK-DAG: %[[C0:.+]] = arith.constant 0 : index
473+
// CHECK-DAG: %[[C1:.+]] = arith.constant 1 : index
474+
// CHECK-DAG: %[[C2:.+]] = arith.constant 2 : index
475+
// CHECK-DAG: %[[C3:.+]] = arith.constant 3 : index
476476
// CHECK: %[[DIM:.+]] = tensor.dim %[[ARG0]], %[[C0]] : tensor<?x?x?x?xf32>
477477
// CHECK: %[[DIM0:.+]] = tensor.dim %[[ARG0]], %[[C1]] : tensor<?x?x?x?xf32>
478478
// CHECK: %[[DIM1:.+]] = tensor.dim %[[ARG0]], %[[C2]] : tensor<?x?x?x?xf32>
@@ -509,8 +509,8 @@ func.func @linalg_transpose_tensor_pack_multiple_tiles(%arg0: tensor<?x32x128xbf
509509
// CHECK: #[[map:.+]] = affine_map<()[s0] -> (s0 ceildiv 16)>
510510
//CHECK-LABEL: func.func @linalg_transpose_tensor_pack_multiple_tiles(
511511
// CHECK-SAME: %[[ARG0:.+]]: tensor<?x32x128xbf16>) -> tensor<32x?x64x16x2xbf16> {
512-
// CHECK: %[[C0:.+]] = arith.constant 0 : index
513-
// CHECK: %[[CST:.+]] = arith.constant 0.000000e+00 : bf16
512+
// CHECK-DAG: %[[C0:.+]] = arith.constant 0 : index
513+
// CHECK-DAG: %[[CST:.+]] = arith.constant 0.000000e+00 : bf16
514514
// CHECK: %[[DIM:.+]] = tensor.dim %[[ARG0]], %[[C0]] : tensor<?x32x128xbf16>
515515
// CHECK: %[[VAL0:.+]] = affine.apply #[[map:.+]]()[%[[DIM]]]
516516
// CHECK: %[[VAL1:.+]] = tensor.empty(%[[VAL0]]) : tensor<32x?x64x16x2xbf16>

0 commit comments

Comments
 (0)