[Mlir-commits] [mlir] [mlir][linalg] Extend elementwise (PR #124661)

[Andrzej Warzyński]

================
@@ -551,6 +551,122 @@ def BroadcastOp : LinalgStructuredBase_Op<"broadcast", [
   let hasCanonicalizer = 1;
 }
 
+//===----------------------------------------------------------------------===//
+// Op definition for ElementwiseOp
+//===----------------------------------------------------------------------===//
+def ElementwiseOp : LinalgStructuredBase_Op<"elementwise", [
+                   AttrSizedOperandSegments]> {
+  let summary = [{ Performs element-wise operation }];
+  let description = [{
+    Linalg op form which performs element-wise computation.
+
+    The attribute `kind` describes the operation (e.g. add, exp). The operation
+    kind can be any elementwise nary (e.g. unary, binary) operation.
+
+    Affine-maps for operands and result are required to be provided by the user
+    when transpose and/or broadcast is needed on any operand. When a map is not
----------------
banach-space wrote:

1. Could you start with the default behaviour (e.g. "By default, all indexing maps are identities.")?
2. Is it OK to specify only one (or two) of the maps? Or is it "either all or nothing"? Please clarify.

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