[Mlir-commits] [mlir] [mlir][vector] Migrate drop-lead-unit-dim to shape_cast (PR #196206)
Andrzej WarzyĆski
llvmlistbot at llvm.org
Fri May 8 04:06:54 PDT 2026
================
@@ -331,6 +393,16 @@ struct CastAwayTransferWriteLeadingOneDim
} // namespace
+namespace {
+struct ContractionOperandDropInfo {
+ AffineMap map;
+ bool needsShapeCast = false;
+ bool transposeNeeded = false;
+ bool transposeNonOuterUnitDims = false;
+ SmallVector<int64_t> permutation;
+};
----------------
banach-space wrote:
* What is `ContractionOperandDropInfo` meant to mean? Drop what?
* `needsShapeCast` vs `transposeNeeded` - these names are inconsistent.
In general, I am confused by this naming. IIUC, `needsShapeCast` encodes _how_ to drop a leading unit dim, right? But that's an implementation detail. The high-level info that's important is whether we need to drop a leading unit dim, right?
https://github.com/llvm/llvm-project/pull/196206
More information about the Mlir-commits
mailing list