This is an archive of the discontinued LLVM Phabricator instance.

Differential D153357

[mlir][linalg][NFC] Add test case for memref vectorization
ClosedPublic

Authored by springerm on Jun 20 2023, 7:57 AM.

Details

Reviewers
nicolasvasilache
aartbik
Commits
rG42a82b1ac6c8: [mlir][linalg][NFC] Add test case for memref vectorization
Summary

Add test cases for vectorizing linalg.matmul and linalg.copy on tensors.

Diff Detail

Event Timeline

springerm created this revision.Jun 20 2023, 7:57 AM
Herald added a reviewer: aartbik. · View Herald TranscriptJun 20 2023, 7:57 AM
Herald added a project: Restricted Project. · View Herald Transcript
Herald added subscribers: bviyer, Moerafaat, bzcheeseman and 24 others. · View Herald Transcript
springerm requested review of this revision.Jun 20 2023, 7:57 AM
Herald added a project: Restricted Project. · View Herald TranscriptJun 20 2023, 7:57 AM
Herald added subscribers: limo1996, stephenneuendorffer. · View Herald Transcript
nicolasvasilache accepted this revision.Jun 20 2023, 8:54 AM
This revision is now accepted and ready to land.Jun 20 2023, 8:54 AM
Harbormaster completed remote builds in B240024: Diff 532937.Jun 20 2023, 9:29 AM
Closed by commit rG42a82b1ac6c8: [mlir][linalg][NFC] Add test case for memref vectorization (authored by springerm). · Explain WhyJun 21 2023, 12:23 AM
This revision was automatically updated to reflect the committed changes.
springerm added a commit: rG42a82b1ac6c8: [mlir][linalg][NFC] Add test case for memref vectorization.

Revision Contents

PathSize
mlir/
test/
Dialect/
Linalg/
45 lines

Diff 533160

mlir/test/Dialect/Linalg/transform-op-vectorize.mlir

Show All 19 Lines
^bb1(%arg1: !transform.any_op): ^bb1(%arg1: !transform.any_op):
%0 = transform.structured.match ops{["linalg.matmul"]} in %arg1 : (!transform.any_op) -> !transform.any_op %0 = transform.structured.match ops{["linalg.matmul"]} in %arg1 : (!transform.any_op) -> !transform.any_op
%1 = get_closest_isolated_parent %0 : (!transform.any_op) -> !transform.any_op %1 = get_closest_isolated_parent %0 : (!transform.any_op) -> !transform.any_op
%2 = transform.structured.vectorize %1 : (!transform.any_op) -> !transform.any_op %2 = transform.structured.vectorize %1 : (!transform.any_op) -> !transform.any_op
} }
// ----- // -----
// CHECK-LABEL: @vectorize_matmul_memref
// CHECK-SAME: %[[A:.*]]: memref<24x12xf32>
// CHECK-SAME: %[[B:.*]]: memref<12x25xf32>
// CHECK-SAME: %[[C:.*]]: memref<24x25xf32>
func.func @vectorize_matmul_memref(%arg0: memref<24x12xf32>,
%arg1: memref<12x25xf32>,
%arg2: memref<24x25xf32>) {
// CHECK: %[[vA:.+]] = vector.transfer_read %[[A]]
// CHECK: %[[vB:.+]] = vector.transfer_read %[[B]]
// CHECK: %[[vC:.+]] = vector.transfer_read %[[C]]
// CHECK: %[[vR:.+]] = vector.contract {{.*}} %[[vA]], %[[vB]], %[[vC]]
// CHECK: vector.transfer_write %[[vR]], %[[C]]
linalg.matmul ins(%arg0, %arg1 : memref<24x12xf32>, memref<12x25xf32>) outs(%arg2 : memref<24x25xf32>)
return
}
transform.sequence failures(propagate) {
^bb1(%arg1: !transform.any_op):
%0 = transform.structured.match ops{["linalg.matmul"]} in %arg1 : (!transform.any_op) -> !transform.any_op
%1 = get_closest_isolated_parent %0 : (!transform.any_op) -> !transform.any_op
%2 = transform.structured.vectorize %1 : (!transform.any_op) -> !transform.any_op
}
// -----
// CHECK-LABEL: @vectorize_copy_memref
// CHECK-SAME: %[[A:.*]]: memref<100x100xf32>,
// CHECK-SAME: %[[B:.*]]: memref<100x100xf32>
func.func @vectorize_copy_memref(%arg0: memref<100x100xf32>,
%arg1: memref<100x100xf32>) {
// CHECK: %[[vA:.+]] = vector.transfer_read %[[A]]
// CHECK: vector.transfer_write %[[vA]], %[[B]]
linalg.copy ins(%arg0 : memref<100x100xf32>) outs(%arg1 : memref<100x100xf32>)
return
}
transform.sequence failures(propagate) {
^bb1(%arg1: !transform.any_op):
%0 = transform.structured.match ops{["linalg.copy"]} in %arg1 : (!transform.any_op) -> !transform.any_op
%1 = get_closest_isolated_parent %0 : (!transform.any_op) -> !transform.any_op
%2 = transform.structured.vectorize %1 : (!transform.any_op) -> !transform.any_op
}
// -----
#map0 = affine_map<()[s0] -> (-s0 + 12, 7)> #map0 = affine_map<()[s0] -> (-s0 + 12, 7)>
#map1 = affine_map<()[s0] -> (-s0 + 7)> #map1 = affine_map<()[s0] -> (-s0 + 7)>
// CHECK-LABEL: @vectorize_keep_pad // CHECK-LABEL: @vectorize_keep_pad
// CHECK-SAME: %[[C:[a-zA-Z0-9_]+]]: tensor<24x25xf32> // CHECK-SAME: %[[C:[a-zA-Z0-9_]+]]: tensor<24x25xf32>
func.func @vectorize_keep_pad( func.func @vectorize_keep_pad(
%arg0: tensor<24x12xf32>, %arg1: tensor<12x25xf32>, %arg0: tensor<24x12xf32>, %arg1: tensor<12x25xf32>,
%arg2: tensor<24x25xf32>, %arg3: index, %arg4: index, %arg2: tensor<24x25xf32>, %arg3: index, %arg4: index,
▲ Show 20 Lines • Show All 100 Lines • Show Last 20 Lines