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

ttnn.squeeze unable to handle tiled inputs #13227

Closed
yugi957 opened this issue Sep 27, 2024 · 0 comments
Closed

ttnn.squeeze unable to handle tiled inputs #13227

yugi957 opened this issue Sep 27, 2024 · 0 comments
Assignees
Labels
bug Something isn't working Op Generalization Generalization and relaxations of requirements in Ops P1 pytorch-compiler

Comments

@yugi957
Copy link
Contributor

yugi957 commented Sep 27, 2024

Describe the bug
(core.cpp:49: tt::exception\ninfo:\nUnable to reshape a tensor in TILE_LAYOUT to non-tile height and width! Please convert the tensor to ROW_MAJOR_LAYOUT first.)
Squeeze can't handle tiled inputs.

To Reproduce
Steps to reproduce the behavior (make sure environment is built and you have access to elastic cloud/corporate):

  1. Go to 'tt_metal/tests/sweep_framework'
  2. Run 'python3 parameter_generator.py --elastic <cloud/corporate> --module-name sweeps.data_movement.squeeze.squeeze_pytorch2' to generate tensor parameters
  3. Run 'python3 runner.py --elastic <cloud/corporate> --module-name sweeps.data_movement.squeeze.squeeze_pytorch2' to run tests
  4. Go to Elastic DB dashboard to view recent sweep run and filter exceptions by the above exception.

Expected behavior
Should run with no errors and output the squeezed tensor exactly as PyTorch does.

Additional context
This was about 35% of total squeeze fails.

@yugi957 yugi957 added bug Something isn't working P1 Op Generalization Generalization and relaxations of requirements in Ops labels Sep 27, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working Op Generalization Generalization and relaxations of requirements in Ops P1 pytorch-compiler
Projects
Status: Done
Development

No branches or pull requests

3 participants