[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