[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