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

Fix squeeze Dim == 0 with padding #12940

Merged
merged 1 commit into from
Sep 20, 2024
Merged

Conversation

ntarafdar
Copy link
Contributor

@ntarafdar ntarafdar commented Sep 20, 2024

Ticket

Link to Github Issue

Problem description

Regressed on this. Was handled in python before and that handling was removed. Now doing the same stuff in C++.

What's changed

Handling dim == 0 to take into consideration padding in squeeze

Checklist

  • Post commit CI passes
  • Blackhole Post commit (if applicable)
  • Model regression CI testing passes (if applicable)
  • Device performance regression CI testing passes (if applicable)
  • New/Existing tests provide coverage for changes

Copy link
Contributor

@TT-BrianLiu TT-BrianLiu left a comment

Choose a reason for hiding this comment

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

Minor comment

ttnn/cpp/ttnn/operations/data_movement/squeeze/squeeze.cpp Outdated Show resolved Hide resolved
@ntarafdar ntarafdar merged commit 613505d into main Sep 20, 2024
6 checks passed
@ntarafdar ntarafdar deleted the tarafdarTT/unsqueeze_dim_0 branch September 20, 2024 21:12
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.

5 participants