Skip to content

Commit d6162ea

Browse files
committed
Add leading scalable dim test
1 parent 6f3b57f commit d6162ea

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

mlir/test/Dialect/Vector/vector-bitcast-lowering-transforms.mlir

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,17 @@ func.func @vector_bitcast_2d_trailing_scalable_dim(%arg0: vector<2x[2]xi64>) ->
6262
// CHECK: %[[B2:.+]] = vector.bitcast %[[V2]] : vector<[2]xi64> to vector<[4]xi32>
6363
// CHECK: %[[R2:.+]] = vector.insert %[[B2]], %[[R1]] [1] : vector<[4]xi32> into vector<2x[4]xi32>
6464
// CHECK: return %[[R2]] : vector<2x[4]xi32>
65+
66+
func.func @vector_bitcast_2d_leading_scalable_dim(%arg0: vector<[2]x2xi64>) -> vector<[2]x4xi32>
67+
{
68+
%0 = vector.bitcast %arg0 : vector<[2]x2xi64> to vector<[2]x4xi32>
69+
return %0 : vector<[2]x4xi32>
70+
}
71+
// CHECK-LABEL: func.func @vector_bitcast_2d_leading_scalable_dim
72+
// CHECK-SAME: %[[IN:[a-zA-Z0-9]+]]
73+
// CHECK: %[[RES:.+]] = vector.bitcast %[[IN]] : vector<[2]x2xi64> to vector<[2]x4xi32>
74+
// CHECK: return %[[RES]] : vector<[2]x4xi32>
75+
6576
module attributes {transform.with_named_sequence} {
6677
transform.named_sequence @__transform_main(%module_op: !transform.any_op {transform.readonly}) {
6778
%f = transform.structured.match ops{["func.func"]} in %module_op

0 commit comments

Comments
 (0)