Skip to content

Commit b46a781

Browse files
committed
fixup! [mlir][nfc] Update vectorize-tensor-extract.mlir (2/N)
Restore CHECK-DAG
1 parent bad48ea commit b46a781

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mlir/test/Dialect/Linalg/vectorize-tensor-extract.mlir

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -98,10 +98,10 @@ func.func @vectorize_nd_tensor_extract_scalar_broadcast(%src: tensor<3x3xf32>, %
9898
// CHECK-LABEL: func.func @vectorize_nd_tensor_extract_scalar_broadcast(
9999
// CHECK-SAME: %[[SRC:.*]]: tensor<3x3xf32>,
100100
// CHECK-SAME: %[[INIT:.*]]: tensor<1x1x3xf32>) -> tensor<1x1x3xf32> {
101-
// CHECK: %[[C0:.*]] = arith.constant 0 : index
102-
// CHECK: %[[C1:.*]] = arith.constant 1 : index
103-
// CHECK: %[[C2:.*]] = arith.constant 2 : index
104-
// CHECK: %[[PAD:.*]] = arith.constant 0.000000e+00 : f32
101+
// CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index
102+
// CHECK-DAG: %[[C1:.*]] = arith.constant 1 : index
103+
// CHECK-DAG: %[[C2:.*]] = arith.constant 2 : index
104+
// CHECK-DAG: %[[PAD:.*]] = arith.constant 0.000000e+00 : f32
105105
// CHECK: %[[READ:.*]] = vector.transfer_read %[[SRC]][%[[C1]], %[[C2]]], %[[PAD]] : tensor<3x3xf32>, vector<f32>
106106
// CHECK: %[[READ_BCAST:.*]] = vector.broadcast %[[READ]] : vector<f32> to vector<1x1x3xf32>
107107
// CHECK: vector.transfer_write %[[READ_BCAST]], %[[INIT]][%[[C0]], %[[C0]], %[[C0]]] {in_bounds = [true, true, true]} : vector<1x1x3xf32>, tensor<1x1x3xf32>

0 commit comments

Comments
 (0)