[Mlir-commits] [mlir] [mlir][linalg] Extend elementwise (PR #124661)
Renato Golin
llvmlistbot at llvm.org
Tue Jan 28 08:22:33 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
+ kind can be any elementwise nary (e.g. unary, binary) operation.
+
+ Affine-maps for operands and result are reuired to be provided by the user
----------------
rengolin wrote:
```suggestion
Affine-maps for operands and result are required to be provided by the user
```
https://github.com/llvm/llvm-project/pull/124661
More information about the Mlir-commits
mailing list