tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.
Apache License 2.0
386 stars 47 forks source link

Move Repeat_interleave #9750

Closed ayerofieiev-tt closed 1 month ago

ayerofieiev-tt commented 2 months ago

tt_eager --> ttnn per op plan

We propose next order to breakdown this work into smaller pieces:

  1. Replace usage of a given op in C++ with ttnn analog (example)
  2. Replace usage in Python [Models, Demos, Examples, tests] (example)
  3. Remove tt_lib bindings and entries from docs (bindings, docs)
  4. Restructure op consolidating code in ttnn (see new structure)

Replacing usage in C++

Each of such usage should be replaced with a ttnn analog. For example repeat --> ttnn::repeat. This should be done for each operation. Missing operations should be added to ttnn.

Replacing usage in Python

For every unary op, look for next entries in Tests/Sweeps, Demos, Models, Examples:

⚠️ tt_lib operations might sometimes have a slightly different interface

Testing

For the best coverage, I recommend to run these workflows. If some of them fails, check if it is the same fail as on main:

ayerofieiev-tt commented 1 month ago

Discussed, we want it as the Op in the data movement folder

KalaivaniMCW commented 1 month ago

Merged to main #10719