Add StableHLO to TTIR conversion for gather OP

tenstorrent / tt-mlir

Tenstorrent MLIR compiler

https://tenstorrent.github.io/tt-mlir/

Apache License 2.0

75 stars 11 forks source link

Open mrakitaTT opened 3 weeks ago

mrakitaTT commented 3 weeks ago

Add support for StableHLO to TTIR conversion for gather OP, lower to ttnn and add unit tests to tt-torch and tt-xla.

We can use ttnn.embedding for simple cases, need new op otherwise.

mrakitaTT commented 3 weeks ago

PR https://github.com/tenstorrent/tt-mlir/pull/938 adds support for cases where we can lower to ttnn.embedding op.