@@ -941,20 +941,17 @@ module attributes {transform.with_named_sequence} {
941
941
942
942
// CHECK-LABEL: func @test_vectorize_dynamic_shapes_unpack
943
943
// CHECK-SAME: %[[ARG_0:.*]]: tensor<?x?xf32>,
944
+ // CHECK-SAME: %[[ARG_1:.*]]: tensor<?x?x16x2xf32>
944
945
func.func @test_vectorize_dynamic_shapes_unpack (%arg0: tensor <?x?xf32 >, %arg1: tensor <?x?x16 x2 xf32 >) -> tensor <?x?xf32 > {
945
946
// CHECK: %[[C0:.*]] = arith.constant 0
946
- // CHECK: %[[DIM:.*]] = tensor.dim %arg0, %[[C0]] : tensor<?x?xf32>
947
- // CHECK: %[[C1:.*]] = arith.constant 1 : index
948
- // CHECK: %[[DIM0:.*]] = tensor.dim %arg0, %[[C1]] : tensor<?x?xf32>
949
- // CHECK: %[[CST:.*]] = arith.constant 0.000000e+00
950
947
// CHECK: %[[C01:.*]] = arith.constant 0
951
948
// CHECK: %[[C02:.*]] = arith.constant 0
952
- // CHECK: %[[DIM4 :.*]] = tensor.dim %arg1 , %[[C02]] : tensor<?x?x16x2xf32>
953
- // CHECK: %[[CNST14 :.*]] = arith.constant 1
954
- // CHECK: %[[DIM6:.*]] = tensor.dim %arg1 , %[[CNST14 ]] : tensor<?x?x16x2xf32>
949
+ // CHECK: %[[DIM_0 :.*]] = tensor.dim %[[ARG_1]] , %[[C02]] : tensor<?x?x16x2xf32>
950
+ // CHECK: %[[C1 :.*]] = arith.constant 1
951
+ // CHECK: %[[DIM6:.*]] = tensor.dim %[[ARG_1]] , %[[C1 ]] : tensor<?x?x16x2xf32>
955
952
// CHECK: %[[CNST16:.*]] = arith.constant 16 : index
956
953
// CHECK: %[[CNST2:.*]] = arith.constant 2 : index
957
- // CHECK: %[[readMsk0:.*]] = vector.create_mask %[[DIM4 ]], %[[DIM6]], %[[CNST16]], %[[CNST2]] : vector<2x1x16x2xi1>
954
+ // CHECK: %[[readMsk0:.*]] = vector.create_mask %[[DIM_0 ]], %[[DIM6]], %[[CNST16]], %[[CNST2]] : vector<2x1x16x2xi1>
958
955
// CHECK: %[[read0:.*]] = vector.mask %[[readMsk0]] {{.*}} vector.transfer_read %{{.*}} : tensor<?x?x16x2xf32>, vector<2x1x16x2xf32> } : vector<2x1x16x2xi1> -> vector<2x1x16x2xf32>
959
956
// CHECK: %[[trans0:.*]] = vector.transpose %[[read0]], [0, 3, 1, 2] : vector<2x1x16x2xf32> to vector<2x2x1x16xf32>
960
957
// CHECK: %[[sc0:.*]] = vector.shape_cast %[[trans0]] : vector<2x2x1x16xf32> to vector<4x16xf32>
0 commit comments