diff options
author | Andrzej Warzyński <andrzej.warzynski@arm.com> | 2025-08-01 11:26:19 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2025-08-01 11:26:19 +0100 |
commit | 77363fbd7ca24206a991eeb43d5ded54dd52a1f6 (patch) | |
tree | 980bdc171c76f94769817456f2d880a8d4b1437b /lldb/unittests/ScriptInterpreter/Python | |
parent | ebe6eba62580592af7065a36b22d929c15291e9a (diff) | |
download | llvm-77363fbd7ca24206a991eeb43d5ded54dd52a1f6.zip llvm-77363fbd7ca24206a991eeb43d5ded54dd52a1f6.tar.gz llvm-77363fbd7ca24206a991eeb43d5ded54dd52a1f6.tar.bz2 |
[mlir][linalg] Add getCollapsedVecType and update vectorization of linalg.unpack (#151503)
This patch introduces a new helper, `getCollapsedVecType`, and updates
`vectorizeAsTensorUnpackOp` to use it. The motivation stems from improving how
`vector.shape_cast` operations are generated when vectorizing `linalg.unpack`.
Previously, the vectorizer relied on
* `tensor::CollapseShapeOp::inferCollapsedType`
to compute the collapsed vector type. This approach is suboptimal
because:
* `inferCollapsedType` lacks awareness of scalable vector flags.
* Linalg vectorization should not depend on Tensor dialect utilities.
Instead of relocating `inferCollapsedType`, we introduce
`getCollapsedVecType` — a lightweight, specialized hook that:
* Assumes no dynamic sizes.
* Handles scalable flags alongside shape dimensions.
This change also reduces temporary variables in
`vectorizeAsTensorUnpackOp` and paves the way for a cleaner update in
#149293.
Diffstat (limited to 'lldb/unittests/ScriptInterpreter/Python')
0 files changed, 0 insertions, 0 deletions