@@ -37,12 +37,11 @@ func.func @add_tensor_test1(%arg0: tensor<4x8x16xf32>, %arg1: tensor<4x8x16xf32>
37
37
// CHECK: %[[C0:.*]] = arith.constant 0 : index
38
38
// CHECK: %[[READ0:.*]] = vector.transfer_read %{{.*}} {in_bounds = [true, true, true]} : tensor<4x16x16xf32>, vector<4x16x16xf32>
39
39
// CHECK: %[[SHAPECAST0:.*]] = vector.shape_cast %[[READ0]] : vector<4x16x16xf32> to vector<4x4x4x4x4xf32>
40
- // CHECK: %[[TRANSPOSE0:.*]] = vector.transpose %[[SHAPECAST0]], [1, 0, 3, 2, 4] : vector<4x4x4x4x4xf32> to vector<4x4x4x4x4xf32>
41
40
// CHECK: %[[READ1:.*]] = vector.transfer_read %{{.*}} {in_bounds = [true, true, true]} : tensor<4x16x16xf32>, vector<4x16x16xf32>
42
41
// CHECK: %[[SHAPECAST1:.*]] = vector.shape_cast %[[READ1]] : vector<4x16x16xf32> to vector<4x4x4x4x4xf32>
43
- // CHECK: %[[TRANSPOSE1:.*]] = vector.transpose %[[SHAPECAST1]], [1, 0, 3, 2, 4] : vector<4x4x4x4x4xf32> to vector<4x4x4x4x4xf32>
44
42
// CHECK: %[[EMPTY:.*]] = tensor.empty() : tensor<4x4x4x4x4xf32>
45
- // CHECK: %[[ADD0:.*]] = arith.addf %[[TRANSPOSE0]], %[[TRANSPOSE1]] : vector<4x4x4x4x4xf32>
43
+ // CHECK: %[[ADD0:.*]] = arith.addf %[[SHAPECAST0]], %[[SHAPECAST1]] : vector<4x4x4x4x4xf32>
44
+ // CHECK: %[[TRANSPOSE1:.*]] = vector.transpose %[[ADD0]], [1, 0, 3, 2, 4] : vector<4x4x4x4x4xf32> to vector<4x4x4x4x4xf32>
46
45
// CHECK: %[[WRITE0:.*]] = vector.transfer_write %{{.*}} {in_bounds = [true, true, true, true, true]} : vector<4x4x4x4x4xf32>, tensor<4x4x4x4x4xf32>
47
46
func.func @add_tensor_pack_test2 (%arg0: tensor <4 x16 x16 xf32 >, %arg1: tensor <4 x16 x16 xf32 >) -> tensor <4 x4 x4 x4 x4 xf32 > {
48
47
%0 = tensor.empty () : tensor <4 x4 x4 x4 x4 xf32 >
0 commit comments