Skip to content

Commit b0958bd

Browse files
authored
[mlir][Vector] Add missing test (scalable vec + strided access) (#81296)
This is a follow-up for #81187, it simply adds missing tests for scalable vectors.
1 parent 0bf4ff2 commit b0958bd

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

mlir/test/Dialect/Vector/vector-dropleadunitdim-transforms.mlir

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -248,6 +248,16 @@ func.func @cast_away_insert_strided_slice_leading_one_dims_one_element(%arg0: ve
248248
return %0: vector<1x1x1xf16>
249249
}
250250

251+
// CHECK-LABEL: func @cast_away_insert_strided_slice_leading_one_dims_one_element_scalable
252+
// CHECK-SAME: %[[ARG0:.+]]: vector<1x[1]xf16>, %{{.+}}: vector<1x1x[1]xf16>
253+
func.func @cast_away_insert_strided_slice_leading_one_dims_one_element_scalable(%arg0: vector<1x[1]xf16>, %arg1: vector<1x1x[1]xf16>) -> vector<1x1x[1]xf16> {
254+
// CHECK: %[[EXT:.+]] = vector.extract %{{.*}}[0] : vector<[1]xf16> from vector<1x[1]xf16>
255+
// CHECK: %[[B:.+]] = vector.broadcast %[[EXT]] : vector<[1]xf16> to vector<1x1x[1]xf16>
256+
%0 = vector.insert_strided_slice %arg0, %arg1 {offsets = [0, 0, 0], strides = [1, 1]} : vector<1x[1]xf16> into vector<1x1x[1]xf16>
257+
// CHECK: return %[[B]]
258+
return %0: vector<1x1x[1]xf16>
259+
}
260+
251261
// CHECK-LABEL: func @cast_away_transfer_read_leading_one_dims
252262
func.func @cast_away_transfer_read_leading_one_dims(%arg0: memref<1x4x8x16xf16>) -> vector<1x4xf16> {
253263
// CHECK: %[[C0:.+]] = arith.constant 0 : index

0 commit comments

Comments
 (0)