[Mlir-commits] [mlir] [mlir] Add inferContractionDims util for indexing map inputs (PR #76081)

[lorenzo chelini]

================
@@ -62,6 +62,8 @@ struct ContractionDimensions {
 /// `k`, indices are returned in sorted order.
 /// Returns a failure if any of `m`, `n` or `k` is empty.
 FailureOr<ContractionDimensions> inferContractionDims(LinalgOp linalgOp);
+FailureOr<ContractionDimensions>
----------------
chelini wrote:

Ok, thanks. I don't have a strong opinion here, but I would expect something more generic if we go down to affine maps (i.e., it is unclear why there is a restriction on the number of affine maps). 

https://github.com/llvm/llvm-project/pull/76081