Skip to content

Commit 9927a19

Browse files
committed
Add test
1 parent fb7ddac commit 9927a19

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

mlir/test/Conversion/VectorToArmSME/unsupported.mlir

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,18 @@ func.func @transfer_write_2d__out_of_bounds(%vector : vector<[4]x[4]xf32>, %dest
145145
return
146146
}
147147

148+
// -----
149+
150+
// CHECK-LABEL: func.func @transfer_write_slice_unsupported_permutation
151+
// CHECK-NOT: arm_sme.store_tile_slice
152+
func.func @transfer_write_slice_unsupported_permutation(%vector: vector<[4]x[4]xf32>, %dest : memref<?x?xf32>, %slice_index: index) {
153+
%c0 = arith.constant 0 : index
154+
%slice = vector.extract %vector[%slice_index] : vector<[4]xf32> from vector<[4]x[4]xf32>
155+
vector.transfer_write %slice, %dest[%slice_index, %c0] { permutation_map = affine_map<(d0, d1) -> (d0)>, in_bounds = [true] }: vector<[4]xf32>, memref<?x?xf32>
156+
return
157+
}
158+
159+
148160
//===----------------------------------------------------------------------===//
149161
// vector.outerproduct
150162
//===----------------------------------------------------------------------===//

mlir/test/Conversion/VectorToArmSME/vector-to-arm-sme.mlir

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -378,8 +378,6 @@ func.func @transfer_write_vertical_slice(%vector: vector<[4]x[4]xf32>, %dest : m
378378
return
379379
}
380380

381-
// -----
382-
383381
//===----------------------------------------------------------------------===//
384382
// vector.broadcast
385383
//===----------------------------------------------------------------------===//

0 commit comments

Comments
 (0)