[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