[Mlir-commits] [mlir] [mlir][vector] Fix vector.broadcast lowering for scalable vectors (PR #66344)
Cullen Rhodes
llvmlistbot at llvm.org
Thu Sep 14 02:53:13 PDT 2023
================
@@ -85,7 +85,8 @@ class BroadcastOpLowering : public OpRewritePattern<vector::BroadcastOp> {
if (srcRank < dstRank) {
// Duplication.
VectorType resType =
- VectorType::get(dstType.getShape().drop_front(), eltType);
+ VectorType::get(dstType.getShape().drop_front(), eltType,
+ dstType.getScalableDims().drop_front());
----------------
c-rhodes wrote:
I noticed @MacDue has been doing:
```
VectorType::Builder(dstType).dropDim(0));
```
in his patches, which seems a bit nicer?
https://github.com/llvm/llvm-project/pull/66344
More information about the Mlir-commits
mailing list