[PATCH] D153422: [mlir][Linalg] Add a softmax op

[Quentin Colombet via Phabricator via llvm-commits]

qcolombet created this revision.
qcolombet added reviewers: rengolin, harsh, mravishankar, nicolasvasilache, ftynse, chelini.
qcolombet added a project: MLIR.
Herald added subscribers: bviyer, Moerafaat, bzcheeseman, awarzynski, sdasgup3, wenzhicui, wrengr, jsetoain, hanchung, cota, teijeong, rdzhabarov, tatianashp, msifontes, jurahul, Kayjukh, grosul1, Joonsoo, liufengdb, aartbik, mgester, arpith-jacob, antiagainst, shauheen, rriddle, mehdi_amini.
Herald added a project: All.
qcolombet requested review of this revision.
Herald added subscribers: limo1996, stephenneuendorffer.

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.
@nicolasvasilache co-authored this patch.


Repository:
  rG LLVM Github Monorepo

https://reviews.llvm.org/D153422

Files:
  mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.td
  mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
  mlir/test/Dialect/Linalg/invalid.mlir
  mlir/test/Dialect/Linalg/roundtrip.mlir

-------------- next part --------------
A non-text attachment was scrubbed...
Name: D153422.533235.patch
Type: text/x-patch
Size: 6185 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20230621/10b67743/attachment.bin>