[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