[Mlir-commits] [mlir] [mlir][linalg] Fix vectorization of tensor.extract (PR #118105)
Andrzej Warzyński
llvmlistbot at llvm.org
Wed Dec 4 08:39:42 PST 2024
================
@@ -1165,8 +1165,18 @@ vectorizeTensorExtract(RewriterBase &rewriter, VectorizationState &state,
loc, resultType, extractOp.getTensor(), transferReadIdxs,
permutationMap, inBounds);
+ // Mask this broadcasting xfer_read here rather than relying on the generic
+ // path (the generic path assumes identity masking map, which wouldn't be
+ // valid here).
+ SmallVector<int64_t> readMaskShape{1};
----------------
banach-space wrote:
Thanks for the reminder 🙏🏻
https://github.com/llvm/llvm-project/pull/118105
More information about the Mlir-commits
mailing list