[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