[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