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

Consolidate ttnn to/from data movement ops #46

Merged
merged 2 commits into from
Jul 19, 2024

Conversation

kevinwuTT
Copy link
Contributor

  • Fuse patterns of to_device, to_layout, and from_torch to just from_torch
  • Fuse patterns of from_device, to_layout, and to_torch to just to_torch
  • Remove checking of data movement ops in unit tests
  • Instead of checking the nodes' exact positions in the graph, change to checking for their existences and relative positions to other nodes
  • Update README to reflect changes
  • Update mnist and resnet model tests to catch failures
  • Add a warning when a model fails when collecting runtime metrics
  • Fix metrics collection when original outputs are of Tensor type but compiled outputs are missing.

@kevinwuTT kevinwuTT requested a review from ayerofieiev-tt July 19, 2024 15:00
@kevinwuTT kevinwuTT added this pull request to the merge queue Jul 19, 2024
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to no response for status checks Jul 19, 2024
@ayerofieiev-tt ayerofieiev-tt merged commit 5495d4b into main Jul 19, 2024
1 check passed
@ayerofieiev-tt ayerofieiev-tt deleted the kw/reduce_ttnn_data_ops branch July 19, 2024 15:44
@kevinwuTT kevinwuTT mentioned this pull request Jul 22, 2024
17 tasks
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.

2 participants