[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