Skip to content

Commit 3705157

Browse files
committed
Update tests
Comments updated and second test removed as duplicate.
1 parent 2199bee commit 3705157

File tree

1 file changed

+1
-18
lines changed

1 file changed

+1
-18
lines changed

mlir/test/Dialect/Vector/vector-transfer-to-vector-load-store.mlir

Lines changed: 1 addition & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,7 @@ func.func @transfer_2D_masked(%mem : memref<?x?xf32>, %mask : vector<2x4xi1>) ->
392392
return %res : vector<2x4xf32>
393393
}
394394

395-
// transfer_read/write are lowered to vector.load/store
395+
// Masked transfer_read/write inside are NOT lowered to vector.load/store
396396
// CHECK-LABEL: func @masked_transfer_to_load(
397397
// CHECK-SAME: %[[MEM:.*]]: memref<8x8xf32>,
398398
// CHECK-SAME: %[[IDX:.*]]: index,
@@ -410,23 +410,6 @@ func.func @masked_transfer_to_load(%mem : memref<8x8xf32>, %i : index, %mask : v
410410
return %mem : memref<8x8xf32>
411411
}
412412

413-
// n-D results are also supported.
414-
// CHECK-LABEL: func @masked_transfer_2D(
415-
// CHECK-SAME: %[[MEM:.*]]: memref<8x8xf32>,
416-
// CHECK-SAME: %[[IDX:.*]]: index,
417-
// CHECK-SAME: %[[MASK:.*]]: vector<2x4xi1>) -> memref<8x8xf32>
418-
// CHECK-NOT: vector.load
419-
// CHECK-NOT: vector.store
420-
// CHECK: %[[READ:.*]] = vector.mask %[[MASK]] { vector.transfer_read %[[MEM]][%[[IDX]], %[[IDX]]]{{.*}} : memref<8x8xf32>, vector<2x4xf32> } : vector<2x4xi1> -> vector<2x4xf32>
421-
// CHECK: vector.mask %[[MASK]] { vector.transfer_write %[[READ]], %[[MEM]][%[[IDX]], %[[IDX]]]{{.*}}: vector<2x4xf32>, memref<8x8xf32> } : vector<2x4xi1>
422-
423-
func.func @masked_transfer_2D(%mem : memref<8x8xf32>, %i : index, %mask : vector<2x4xi1>) -> memref<8x8xf32> {
424-
%cf0 = arith.constant 0.0 : f32
425-
%read = vector.mask %mask { vector.transfer_read %mem[%i, %i], %cf0 {in_bounds = [true, true]} : memref<8x8xf32>, vector<2x4xf32> } : vector<2x4xi1> -> vector<2x4xf32>
426-
vector.mask %mask {vector.transfer_write %read, %mem[%i, %i] {in_bounds = [true, true]} : vector<2x4xf32>, memref<8x8xf32> } : vector<2x4xi1>
427-
return %mem : memref<8x8xf32>
428-
}
429-
430413
module attributes {transform.with_named_sequence} {
431414
transform.named_sequence @__transform_main(%root : !transform.any_op {transform.readonly}) {
432415
%func_op = transform.structured.match ops{["func.func"]} in %root : (!transform.any_op) -> !transform.op<"func.func">

0 commit comments

Comments
 (0)