Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

#8835: Introduce a faster more flexible operation infra v2 #9279

Merged
merged 5 commits into from
Jun 18, 2024

Conversation

arakhmati
Copy link
Contributor

@arakhmati arakhmati commented Jun 7, 2024

Ticket

#8835

Problem description

Current infrastructure for writing a device operation relies heavily on using std::vectors in the common parts of the infra. They were used because they are easier to understand.
However using vectors causes 2 issues:

  • performance degradation to extra heap allocations
  • less flexible APIs

What's changed

Added new infra for writing device operations based on template meta-programming

Checklist

Post commit CI - passed
Model regression CI testing running

@arakhmati arakhmati changed the title #5389: added TMP-based device operation infra #8835: added TMP-based device operation infra Jun 7, 2024
@arakhmati arakhmati force-pushed the ttnn-8835-added-tmp-device-op-infra branch from 8a97b1c to 6d0b80d Compare June 7, 2024 05:21
@arakhmati arakhmati force-pushed the ttnn-8835-added-tmp-device-op-infra branch from 6d0b80d to 85ff37e Compare June 7, 2024 05:44
Copy link
Contributor

@mo-tenstorrent mo-tenstorrent left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just the hard coded name for V2 device ops needs to be updated

@arakhmati arakhmati force-pushed the ttnn-8835-added-tmp-device-op-infra branch from 1bf4110 to 8439e0c Compare June 18, 2024 21:03
@arakhmati arakhmati force-pushed the ttnn-8835-added-tmp-device-op-infra branch from 8439e0c to 2bd9fd3 Compare June 18, 2024 21:04
@arakhmati arakhmati changed the title #8835: added template metaprogramming based device operation infra #8835: Introduce a faster more flexible operation infra v2 Jun 18, 2024
@arakhmati arakhmati merged commit f5b64f8 into main Jun 18, 2024
74 checks passed
@arakhmati arakhmati deleted the ttnn-8835-added-tmp-device-op-infra branch June 18, 2024 21:52
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.

10 participants