[Mlir-commits] [mlir] [MLIR][Linalg] Add aggregate ops decomposition pass and softmax decom… (PR #97582)
Adam Siemieniuk
llvmlistbot at llvm.org
Thu Jul 4 03:22:48 PDT 2024
================
@@ -2622,6 +2622,11 @@ Value createLinalgReduceSumBody(OpBuilder &b, Location loc, ValueRange args,
/// softmax = z / l
///
FailureOr<DecompositionResult> SoftmaxOp::decomposeOperation(OpBuilder &b) {
+ if (!isa<RankedTensorType>(getInput().getType())) {
----------------
adam-smnk wrote:
I'd use `hasPureTensorSemantics` to avoid errors on mixed semantics.
https://github.com/llvm/llvm-project/pull/97582
More information about the Mlir-commits
mailing list