Clément Fournier 36c3466aef

[mlir][linalg] Fix neutral elt for softmax (#118952) ...

The decomposition of `linalg.softmax` uses `maxnumf`, but the identity
element that is used in the generated code is the one for `maximumf`.
They are not the same, as the identity for `maxnumf` is `NaN`, while the
one of `maximumf` is `-Infty`. This is wrong and prevents the maxnumf
from being folded.

Related to #114595, which fixed the folder for maxnumf.

2025-01-13 15:21:07 +08:00

..

IR

[mlir][linalg] Fix neutral elt for softmax (#118952)

2025-01-13 15:21:07 +08:00

TransformOps

[mlir][Linalg] Allow PartialReductionOpInterface ops in tile_reduction_using_for (#120118)

2024-12-27 13:19:58 +00:00

Transforms

[mlir][Transforms] Support 1:N mappings in ConversionValueMapping (#116524)

2025-01-03 16:11:56 +01:00

Utils

[MLIR][Transform] Hoist Pad generates linalg.transpose (#109669)

2024-09-26 09:33:47 +01:00

CMakeLists.txt

[mlir] Remove Linalg fusion-on-memrefs.

2023-01-30 14:42:00 +01:00