[Mlir-commits] [mlir] [mlir][Linalg] Refine how broadcast dims are treated (PR #99015)
Andrzej WarzyĆski
llvmlistbot at llvm.org
Thu Sep 26 08:16:00 PDT 2024
================
@@ -224,10 +224,10 @@ struct VectorizationState {
/// Masks an operation with the canonical vector mask if the operation needs
/// masking. Returns the masked operation or the original operation if masking
/// is not needed. If provided, the canonical mask for this operation is
- /// permuted using `maybeMaskingMap`.
+ /// permuted using `maybeIndexingMap`.
Operation *
maskOperation(RewriterBase &rewriter, Operation *opToMask, LinalgOp linalgOp,
- std::optional<AffineMap> maybeMaskingMap = std::nullopt);
+ std::optional<AffineMap> maybeIndexingMap = std::nullopt);
----------------
banach-space wrote:
Thanks for reminding me about this!
Right now we are actually passing the indexing map whenever this hook is used - this is just to documenting the state today. So let me leave it as is.
I will send a follow-on PR in which I'll try to clarify the difference between the two maps.
https://github.com/llvm/llvm-project/pull/99015
More information about the Mlir-commits
mailing list