[Mlir-commits] [mlir] [mlir][linalg] Scaled contraction op (PR #203958)

[Adam Siemieniuk]

================
@@ -1218,6 +1218,150 @@ def BatchReduceMatmulOp : LinalgStructuredBase_Op<"batch_reduce_matmul", [
     }];
 }
 
+//===----------------------------------------------------------------------===//
+// Scaled Contract op.
+//===----------------------------------------------------------------------===//
+
+def ScaledContractOp : LinalgStructuredBase_Op<"scaled_contract", [
+    AttrSizedOperandSegments]> {
----------------
adam-smnk wrote:

`scaled_contract` doesn't fit the interface - extra scale inputs and their affine maps, complex body.
As it's only a single op, I'd keep it separate instead of trying to change or propose a new interface.

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