[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