@@ -54,7 +54,6 @@ func.func @permutation_with_mask_xfer_write_scalable(%arg0: vector<4x[8]xi16>, %
54
54
// CHECK-SAME: %[[MASK:.*]]: vector<16xi1>
55
55
// CHECK-NOT: vector.transpose
56
56
// CHECK: %[[RES:.*]] = vector.mask %[[MASK]] { vector.transfer_write %[[ARG_1]], %[[ARG_0]][%[[IDX]], %[[IDX]]] {{.*}} vector<16xf32>, tensor<?x?xf32> } : vector<16xi1> -> tensor<?x?xf32>
57
- // CHECK: return %[[RES]]
58
57
func.func @masked_permutation_xfer_write_fixed_width (%t: tensor <?x?xf32 >, %val: vector <16 xf32 >, %idx: index , %mask: vector <16 xi1 >) -> tensor <?x?xf32 > {
59
58
%r = vector.mask %mask { vector.transfer_write %val , %t[%idx , %idx ] {permutation_map = affine_map <(d0 , d1 ) -> (d0 )>} : vector <16 xf32 >, tensor <?x?xf32 > } : vector <16 xi1 > -> tensor <?x?xf32 >
60
59
return %r : tensor <?x?xf32 >
@@ -66,9 +65,7 @@ func.func @masked_permutation_xfer_write_fixed_width(%t: tensor<?x?xf32>, %val:
66
65
// CHECK-SAME: %[[MASK:.*]]: vector<4x[8]xi1>)
67
66
// CHECK-SAME: -> tensor<?x?x?x?xf32> {
68
67
// CHECK-NOT: vector.transpose
69
- // CHECK: %[[C0:.*]] = arith.constant 0 : index
70
68
// CHECK: %[[R:.*]] = vector.mask %[[MASK]] { vector.transfer_write %[[ARG_0]], %[[ARG_1]][%c0, %c0, %c0, %c0] {in_bounds = [true, true], permutation_map = #[[MAP:.*]]} : vector<4x[8]xi16>, tensor<?x?x?x?xf32> } : vector<4x[8]xi1> -> tensor<?x?x?x?xf32>
71
- // CHECK: return %[[R]] : tensor<?x?x?x?xf32>
72
69
func.func @masked_permutation_xfer_write_scalable (%arg0: vector <4 x[8 ]xi16 >, %t: tensor <?x?x?x?xf32 >, %mask: vector <4 x[8 ]xi1 >) -> tensor <?x?x?x?xf32 > {
73
70
%c0 = arith.constant 0 : index
74
71
%r = vector.mask %mask { vector.transfer_write %arg0 , %t [%c0 , %c0 , %c0 , %c0 ] {in_bounds = [true , true ], permutation_map = affine_map <(d0 , d1 , d2 , d3 ) -> (d1 , d2 )>
@@ -83,7 +80,6 @@ func.func @masked_permutation_xfer_write_scalable(%arg0: vector<4x[8]xi16>, %t:
83
80
// CHECK-SAME: %[[ARG1:.*]]: vector<14x8x16xf32>
84
81
// CHECK-SAME: %[[IDX:.*]]: index) -> tensor<?x?x?x?xf32>
85
82
// CHECK-NOT: vector.broadcast
86
- // CHECK: %[[C0:.*]] = arith.constant 0 : index
87
83
// CHECK: %[[masked1:.*]] = vector.mask %0 { vector.transfer_write %[[ARG1]], %[[ARG0]]{{.*}}permutation_map = #[[MAP:.*]]} : vector<14x8x16xf32>, tensor<?x?x?x?xf32> } : vector<14x8x16xi1> -> tensor<?x?x?x?xf32>
88
84
func.func @masked_non_permutation_xfer_write_fixed_width (
89
85
%arg0 : tensor <?x?x?x?xf32 >,
@@ -169,9 +165,7 @@ func.func private @test.some_use(vector<1x4x4xf32>)
169
165
// CHECK-SAME: %[[ARG_0:.*]]: tensor<?x?xf32>,
170
166
// CHECK-SAME: %[[MASK:.*]]: vector<2x[4]xi1>) -> vector<8x[4]x2xf32> {
171
167
// CHECK-NOT: vector.transpose
172
- // CHECK: %[[C0:.*]] = arith.constant 0 : index
173
168
// CHECK: %[[T_READ:.*]] = vector.mask %[[MASK]] { vector.transfer_read %[[ARG_0]][%[[C0]], %[[C0]]], %cst {in_bounds = [true, true, true], permutation_map = #[[MAP:.*]]} : tensor<?x?xf32>, vector<8x[4]x2xf32> } : vector<2x[4]xi1> -> vector<8x[4]x2xf32>
174
- // CHECK: return %[[T_READ]] : vector<8x[4]x2xf32>
175
169
func.func @masked_permutation_xfer_read_scalable (%t: tensor <?x?xf32 >, %mask : vector <2 x[4 ]xi1 >) -> vector <8 x[4 ]x2 xf32 > {
176
170
177
171
%c0 = arith.constant 0 : index
0 commit comments