@@ -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