Skip to content

Commit 24be4d5

Browse files
authored
[mlir][vector] Add tests for scalable vectors in one-shot-bufferize.mlir (#102361)
1 parent 4c19de9 commit 24be4d5

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

mlir/test/Dialect/Vector/one-shot-bufferize.mlir

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,17 @@ func.func @mask(%t0: tensor<?xf32>, %val: vector<16xf32>, %idx: index, %m0: vect
1212
return %0 : tensor<?xf32>
1313
}
1414

15+
// CHECK-LABEL: func @mask_scalable(
16+
// CHECK-SAME: %[[t0:.*]]: memref<?xf32, strided<[?], offset: ?>>
17+
func.func @mask_scalable(%t0: tensor<?xf32>, %val: vector<[16]xf32>, %idx: index, %m0: vector<[16]xi1>) -> tensor<?xf32> {
18+
// CHECK-NOT: alloc
19+
// CHECK-NOT: copy
20+
// CHECK: vector.mask %{{.*}} { vector.transfer_write %{{.*}}, %[[t0]][%{{.*}}] : vector<[16]xf32>, memref<?xf32, strided<[?], offset: ?>> } : vector<[16]xi1>
21+
%0 = vector.mask %m0 { vector.transfer_write %val, %t0[%idx] : vector<[16]xf32>, tensor<?xf32> } : vector<[16]xi1> -> tensor<?xf32>
22+
// CHECK: return %[[t0]]
23+
return %0 : tensor<?xf32>
24+
}
25+
1526
// -----
1627

1728
// CHECK-ANALYSIS-LABEL: func @non_reading_xfer_write(

0 commit comments

Comments
 (0)