Skip to content

Conversation

@shutovilyaep
Copy link

Registering Unary TTNN OPs

Problem description

Existing TTNN OPs should be registered to Pytorch Dispatcher.

The scope of this PR is Unary OPs only, other groups of OPs are planned to be added in other PRs
[TODO: to update links when PRs are moved to "tenstorrent" organization]:

What's changed

  • Unary wrappers are introduced
  • Most unary TTNN OPs are registered

@shutovilyaep shutovilyaep force-pushed the feat/register_unary_ops branch from 9e73143 to b367849 Compare November 21, 2025 17:22
shutovilya and others added 7 commits November 22, 2025 06:52
…sts. Added functions to transform tensors between real and complex formats. Updated OpTestConfig to include pre- and post-operation transformations. Modified unary operation tests to validate complex support for 'angle' and 'conj' operations.
@shutovilyaep shutovilyaep force-pushed the feat/register_unary_ops branch from a78a1ab to 2484481 Compare November 22, 2025 09:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants