[Mlir-commits] [mlir] [mlir][vector] Fix FlattenGather for scalable vectors (PR #96074)

Benjamin Maxwell llvmlistbot at llvm.org
Thu Jun 20 01:54:42 PDT 2024


================
@@ -73,7 +80,8 @@ struct FlattenGather : OpRewritePattern<vector::GatherOp> {
         loc, resultTy, rewriter.getZeroAttr(resultTy));
 
     Type subTy = VectorType::get(resultTy.getShape().drop_front(),
-                                 resultTy.getElementType());
+                                 resultTy.getElementType(),
----------------
MacDue wrote:

nit: This could be:
```cpp
Type subTy = VectorType::Builder(resultTy).dropDim(0);
```

https://github.com/llvm/llvm-project/pull/96074


More information about the Mlir-commits mailing list