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

Andrzej WarzyƄski llvmlistbot at llvm.org
Mon Feb 3 12:00:14 PST 2025


================
@@ -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
----------------
banach-space wrote:

[nit]
```suggestion
    The attribute `kind` describes the arithmetic operation to perform. This operation
    can either be unary (e.g. max), binary (e.g. add) or ternary (i.e. select).
```

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


More information about the Mlir-commits mailing list