[Mlir-commits] [mlir] [MLIR][Linalg] Add aggregate ops decomposition pass and softmax decom… (PR #97582)

Petr Kurapov llvmlistbot at llvm.org
Fri Jul 5 08:23:18 PDT 2024


================
@@ -30,6 +30,16 @@ class IteratorTypeAttr;
 class LinalgOp;
 class GenericOp;
 
+/// Container for result values of decomposition.
+/// - `decomposedOps` contains operations created by the decomposition that are
+/// returned to the caller for further transformations.
+/// - `decomposedValues` contains the values corresponding to the result of the
+/// aggregate operation.
+struct DecompositionResult {
+  SmallVector<Operation *> decomposedOps;
----------------
kurapov-peter wrote:

In softmax case - yes. I was thinking this could eventually become not a linalg-specific interface, so went with `Operation*`. Do you have a specific reason it should be `LinalgOp` in mind?

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


More information about the Mlir-commits mailing list