diff options
| author | Andrzej Warzynski <andrzej.warzynski@arm.com> | 2023-08-15 16:00:14 +0000 |
|---|---|---|
| committer | Andrzej Warzynski <andrzej.warzynski@arm.com> | 2023-08-22 08:45:59 +0000 |
| commit | 576b184d6e3b633f51b908b61ebd281d2ecbf66f (patch) | |
| tree | b7677ca599c4f781c1c82a9381ab692199ee1263 /mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | |
| parent | 386aa2ab9d19c783deb9fbfb3b9be14754ba8789 (diff) | |
[mlir][vector] Add support for scalable vectors in `trimLeadingOneDims`
This patch updates one specific hook in "VectorDropLeadUnitDim.cpp" to
make sure that "scalable dims" are handled correctly. While this change
affects multiple patterns, I am only adding one regression tests that
captures one specific case that affects me right now.
I am also adding Vector dialect to the list of dependencies of
`-test-vector-to-vector-lowering`. Otherwise my test case won't work as
a standalone test.
Differential Revision: https://reviews.llvm.org/D157993
Diffstat (limited to 'mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp')
| -rw-r--r-- | mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp index 554a7b6db472..370d8bb19364 100644 --- a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp +++ b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp @@ -55,6 +55,7 @@ struct TestVectorToVectorLowering void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<affine::AffineDialect>(); + registry.insert<vector::VectorDialect>(); } Option<bool> unroll{*this, "unroll", llvm::cl::desc("Include unrolling"), |