[all-commits] [llvm/llvm-project] 2d9b91: [mlir][Linalg] Add a softmax op

[qcolombet via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 2d9b9103b45e42feb24058fb1f8e615fdba6ae5c
      https://github.com/llvm/llvm-project/commit/2d9b9103b45e42feb24058fb1f8e615fdba6ae5c
  Author: Quentin Colombet <quentin.colombet at gmail.com>
  Date:   2023-06-29 (Thu, 29 Jun 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.td
    M mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
    M mlir/test/Dialect/Linalg/invalid.mlir
    M mlir/test/Dialect/Linalg/roundtrip.mlir

  Log Message:
  -----------
  [mlir][Linalg] Add a softmax op

This patch adds a softmax op.
For now, nothing interesting happens, we can only do a round trip.
Later patches will add the tiling interface and the lowering of this op to
a sequence of simpler ops.

This is graduating the linag_ext.softmax op from iree to LLVM.

Original implementation from Harsh Menon <harsh at nod-labs.com>
Nicolas Vasilache <nicolas.vasilache at gmail.com> co-authored this patch.

Differential Revision: https://reviews.llvm.org/D153422