Skip to content

Commit c0bd8a9

Browse files
banach-spacetomtor
authored andcommitted
[mlir][linalg] Remove the test-linalg-to-vector-patterns option (llvm#142116)
This patch removes the `test-linalg-to-vector-patterns` option from the `-test-linalg-transform-patterns=` test flag. It was only used in one test, where a more specialized transform dialect op can be used instead: * `transform.apply_patterns.linalg.pad_vectorization` While we could preserve `test-linalg-to-vector-patterns`, it's better to rely on finer-grained transformations — this way, we know exactly what is being run and tested. Now that its only use has been removed, it feels natural to delete `test-linalg-to-vector-patterns`.
1 parent 83e176d commit c0bd8a9

File tree

2 files changed

+12
-17
lines changed

2 files changed

+12
-17
lines changed

mlir/test/Integration/Dialect/Linalg/CPU/test-padtensor.mlir

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
// RUN: mlir-opt %s -test-linalg-transform-patterns=test-linalg-to-vector-patterns \
1+
// RUN: mlir-opt %s -transform-interpreter -test-transform-dialect-erase-schedule \
22
// RUN: -one-shot-bufferize="bufferize-function-boundaries" \
33
// RUN: -buffer-deallocation-pipeline -convert-bufferization-to-memref \
44
// RUN: -convert-linalg-to-loops -convert-scf-to-cf -expand-strided-metadata \
@@ -34,4 +34,15 @@ func.func @main() {
3434
return
3535
}
3636

37+
module attributes {transform.with_named_sequence} {
38+
transform.named_sequence @__transform_main(%arg1: !transform.any_op {transform.readonly}) {
39+
%func_op = transform.structured.match ops{["func.func"]} in %arg1 : (!transform.any_op) -> !transform.op<"func.func">
40+
41+
transform.apply_patterns to %func_op {
42+
transform.apply_patterns.linalg.pad_vectorization
43+
} : !transform.op<"func.func">
44+
transform.yield
45+
}
46+
}
47+
3748
func.func private @printMemrefF32(%ptr : tensor<*xf32>)

mlir/test/lib/Dialect/Linalg/TestLinalgTransforms.cpp

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -65,11 +65,6 @@ struct TestLinalgTransforms
6565
llvm::cl::desc(
6666
"Test a fused pass that forwards memref.copy to vector.transfer"),
6767
llvm::cl::init(false)};
68-
Option<bool> testGenericToVectorPattern{
69-
*this, "test-linalg-to-vector-patterns",
70-
llvm::cl::desc("Test a set of patterns that rewrite a linalg contraction "
71-
"in vector.contract form"),
72-
llvm::cl::init(false)};
7368
Option<bool> testDecomposePadTensor{
7469
*this, "test-decompose-pad-tensor",
7570
llvm::cl::desc("Test transform pad tensor by copying with generic ops"),
@@ -166,15 +161,6 @@ static void applyVectorTransferForwardingPatterns(func::FuncOp funcOp) {
166161
(void)applyPatternsGreedily(funcOp, std::move(forwardPattern));
167162
}
168163

169-
static void applyLinalgToVectorPatterns(func::FuncOp funcOp) {
170-
RewritePatternSet patterns(funcOp.getContext());
171-
auto *ctx = funcOp.getContext();
172-
patterns.add<CopyVectorizationPattern>(ctx);
173-
populatePadOpVectorizationPatterns(patterns);
174-
populateConvolutionVectorizationPatterns(patterns);
175-
(void)applyPatternsGreedily(funcOp, std::move(patterns));
176-
}
177-
178164
static void applyDecomposePadPatterns(func::FuncOp funcOp) {
179165
RewritePatternSet patterns(funcOp.getContext());
180166
patterns.add<DecomposePadOpPattern>(funcOp.getContext());
@@ -254,8 +240,6 @@ void TestLinalgTransforms::runOnOperation() {
254240
return applyPatterns(getOperation());
255241
if (testVectorTransferForwardingPatterns)
256242
return applyVectorTransferForwardingPatterns(getOperation());
257-
if (testGenericToVectorPattern)
258-
return applyLinalgToVectorPatterns(getOperation());
259243
if (testDecomposePadTensor)
260244
return applyDecomposePadPatterns(getOperation());
261245
if (testDecomposeTensorPackOp)

0 commit comments

Comments
 (0)