Skip to content

Commit 1dcd71f

Browse files
committed
fixup! [mlir][nfc] Update vectorize-tensor-extract.mlir (3/N)
Add DAG to FileCHeck prefixes
1 parent adabdf4 commit 1dcd71f

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

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

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -572,10 +572,10 @@ func.func @vectorize_nd_tensor_extract_scalar_broadcast(%src: tensor<3x3xf32>, %
572572
// CHECK-LABEL: func.func @vectorize_nd_tensor_extract_scalar_broadcast(
573573
// CHECK-SAME: %[[SRC:.*]]: tensor<3x3xf32>,
574574
// CHECK-SAME: %[[INIT:.*]]: tensor<1x1x3xf32>) -> tensor<1x1x3xf32> {
575-
// CHECK: %[[C0:.*]] = arith.constant 0 : index
576-
// CHECK: %[[C1:.*]] = arith.constant 1 : index
577-
// CHECK: %[[C2:.*]] = arith.constant 2 : index
578-
// CHECK: %[[PAD:.*]] = arith.constant 0.000000e+00 : f32
575+
// CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index
576+
// CHECK-DAG: %[[C1:.*]] = arith.constant 1 : index
577+
// CHECK-DAG: %[[C2:.*]] = arith.constant 2 : index
578+
// CHECK-DAG: %[[PAD:.*]] = arith.constant 0.000000e+00 : f32
579579
// CHECK: %[[READ:.*]] = vector.transfer_read %[[SRC]][%[[C1]], %[[C2]]], %[[PAD]] : tensor<3x3xf32>, vector<f32>
580580
// CHECK: %[[READ_BCAST:.*]] = vector.broadcast %[[READ]] : vector<f32> to vector<1x1x3xf32>
581581
// CHECK: vector.transfer_write %[[READ_BCAST]], %[[INIT]][%[[C0]], %[[C0]], %[[C0]]] {in_bounds = [true, true, true]} : vector<1x1x3xf32>, tensor<1x1x3xf32>
@@ -621,8 +621,8 @@ func.func @extract_scalar_from_0d_into_1d(%src: tensor<f32>, %init: tensor<1xf32
621621
// CHECK-LABEL: func.func @extract_scalar_from_0d_into_1d(
622622
// CHECK-SAME: %[[SRC:.*]]: tensor<f32>,
623623
// CHECK-SAME: %[[INIT:.*]]: tensor<1xf32>) -> tensor<1xf32> {
624-
// CHECK: %[[C0:.*]] = arith.constant 0 : index
625-
// CHECK: %[[PAD:.*]] = arith.constant 0.000000e+00 : f32
624+
// CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index
625+
// CHECK-DAG: %[[PAD:.*]] = arith.constant 0.000000e+00 : f32
626626
// CHECK: %[[READ:.*]] = vector.transfer_read %[[SRC]][], %[[PAD]] : tensor<f32>, vector<f32>
627627
// CHECK: %[[READ_BCAST:.*]] = vector.broadcast %[[READ]] : vector<f32> to vector<1xf32>
628628
// CHECK: vector.transfer_write %[[READ_BCAST]], %[[INIT]][%[[C0]]] {in_bounds = [true]} : vector<1xf32>, tensor<1xf32>
@@ -670,10 +670,10 @@ func.func @scalar_read_with_broadcast_from_column_tensor(%init: tensor<1x1x4xi32
670670

671671
// CHECK-LABEL: func.func @scalar_read_with_broadcast_from_column_tensor
672672
// CHECK-SAME: %[[INIT:.*]]: tensor<1x1x4xi32>) -> tensor<1x1x4xi32> {
673-
// CHECK: %[[PAD:.*]] = arith.constant 0 : i32
674-
// CHECK: %[[C0:.*]] = arith.constant 0 : index
675-
// CHECK: %[[SRC:.*]] = arith.constant dense<{{\[\[}}0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]]> : tensor<15x1xi32>
676-
// CHECK: %[[IDX_VEC:.*]] = arith.constant dense<0> : vector<1xindex>
673+
// CHECK-DAG: %[[PAD:.*]] = arith.constant 0 : i32
674+
// CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index
675+
// CHECK-DAG: %[[SRC:.*]] = arith.constant dense<{{\[\[}}0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]]> : tensor<15x1xi32>
676+
// CHECK-DAG: %[[IDX_VEC:.*]] = arith.constant dense<0> : vector<1xindex>
677677
// CHECK: %[[IDX_ELT:.*]] = vector.extract %[[IDX_VEC]][0] : index from vector<1xindex>
678678
// CHECK: %[[READ:.*]] = vector.transfer_read %[[SRC]]{{\[}}%[[IDX_ELT]], %[[C0]]], %[[PAD]] : tensor<15x1xi32>, vector<i32>
679679
// CHECK: %[[READ_BCAST:.*]] = vector.broadcast %[[READ]] : vector<i32> to vector<1x1x4xi32>

0 commit comments

Comments
 (0)