[Mlir-commits] [mlir] [MLIR][Linalg] Introduce linalg.contract (PR #123618)

Andrzej WarzyƄski llvmlistbot at llvm.org
Mon Jan 27 05:58:49 PST 2025


================
@@ -680,6 +680,125 @@ def MatmulOp : LinalgStructuredBase_Op<"matmul", [
     }];
 }
 
+//===----------------------------------------------------------------------===//
+// Contract op.
+//===----------------------------------------------------------------------===//
+
+def ContractOp : LinalgStructuredBase_Op<"contract", [
+               AttrSizedOperandSegments,
+               LinalgContractionOpInterface]> {
+  let summary = [{
+    Perform a contraction on two inputs, accumulating on top of a third.
----------------
banach-space wrote:

"Accumulating _into_ the third" would sound more natural to me.

Below you document that C is used as the initial value.

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


More information about the Mlir-commits mailing list