Need support for torch.cumsum in ttnn

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.

Apache License 2.0

480 stars 78 forks source link

Open punithsekar opened 1 day ago

punithsekar commented 1 day ago

Describe

Need support for torch.cumsum operation in ttnn.

Torch documentation Link

Additional Context It is used in SinePositionalEncoding3D class of PETR model.

punithsekar commented 1 day ago

fyi @saichandax