search

tenstorrent / pytorch2.0_ttnn

⭐️ TTNN Compiler for PyTorch 2.0 ⭐️ It enables running PyTorch2.0 models on Tenstorrent hardware
https://tenstorrent.github.io/tt-metal/latest/ttnn/
16 stars 5 forks source link

Test conversion of `torch.flip` #165

Open jdh8 opened 1 week ago

jdh8 commented 1 week ago

Ticket

tenstorrent/tt-metal#12306

Problem description

Lower torch.flip to possibly ttnn.slice

What's changed

We need tt-metal support first, either a directly corresponding ttnn.flip or supporting ttnn.slice(step=-1)