[Mlir-commits] [mlir] [mlir][vector] Fix vector.broadcast lowering for scalable vectors (PR #66344)
Benjamin Maxwell
llvmlistbot at llvm.org
Thu Sep 14 03:38:02 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());
----------------
MacDue wrote:
Yep, that'd be preferred here :+1:
https://github.com/llvm/llvm-project/pull/66344
More information about the Mlir-commits
mailing list