[Mlir-commits] [mlir] [MLIR][Linalg] Introduce broadcast/transpose semantic to 'linalg.batc… (PR #122275)
Md Asghar Ahmad Shahid
llvmlistbot at llvm.org
Mon Jan 13 08:34:49 PST 2025
================
@@ -935,7 +935,8 @@ struct RankReduceContractionOps : OpRewritePattern<FromOpTy> {
loc, collapsedResultTy, ValueRange{collapsedLhs, collapsedRhs},
ValueRange{collapsedInit});
for (auto attr : contractionOp->getAttrs()) {
- if (attr.getName() == LinalgDialect::kMemoizedIndexingMapsAttrName)
+ if (attr.getName() == LinalgDialect::kMemoizedIndexingMapsAttrName ||
+ attr.getName() == "indexing_maps")
----------------
shahidact wrote:
I understand your concern; however, this pass is based on the default behaviour of the `batch_matmul`. In future, if I want to extend this pass for the extended behaviour I would handle it.
https://github.com/llvm/llvm-project/pull/122275
More information about the Mlir-commits
mailing list