@@ -572,10 +572,10 @@ func.func @vectorize_nd_tensor_extract_scalar_broadcast(%src: tensor<3x3xf32>, %
572
572
// CHECK-LABEL: func.func @vectorize_nd_tensor_extract_scalar_broadcast(
573
573
// CHECK-SAME: %[[SRC:.*]]: tensor<3x3xf32>,
574
574
// 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
579
579
// CHECK: %[[READ:.*]] = vector.transfer_read %[[SRC]][%[[C1]], %[[C2]]], %[[PAD]] : tensor<3x3xf32>, vector<f32>
580
580
// CHECK: %[[READ_BCAST:.*]] = vector.broadcast %[[READ]] : vector<f32> to vector<1x1x3xf32>
581
581
// 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
621
621
// CHECK-LABEL: func.func @extract_scalar_from_0d_into_1d(
622
622
// CHECK-SAME: %[[SRC:.*]]: tensor<f32>,
623
623
// 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
626
626
// CHECK: %[[READ:.*]] = vector.transfer_read %[[SRC]][], %[[PAD]] : tensor<f32>, vector<f32>
627
627
// CHECK: %[[READ_BCAST:.*]] = vector.broadcast %[[READ]] : vector<f32> to vector<1xf32>
628
628
// 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
670
670
671
671
// CHECK-LABEL: func.func @scalar_read_with_broadcast_from_column_tensor
672
672
// 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>
677
677
// CHECK: %[[IDX_ELT:.*]] = vector.extract %[[IDX_VEC]][0] : index from vector<1xindex>
678
678
// CHECK: %[[READ:.*]] = vector.transfer_read %[[SRC]]{{\[}}%[[IDX_ELT]], %[[C0]]], %[[PAD]] : tensor<15x1xi32>, vector<i32>
679
679
// CHECK: %[[READ_BCAST:.*]] = vector.broadcast %[[READ]] : vector<i32> to vector<1x1x4xi32>
0 commit comments