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

lorenzo chelini llvmlistbot at llvm.org
Tue Jan 2 14:17:14 PST 2024


================
@@ -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:

Sorry for the latency, I lost this PR. Yes, you are right `ContractionOpInterface` already assumes two inputs and one output. It makes sense to restrict to 3 maps only then.

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


More information about the Mlir-commits mailing list