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.

Sign up for GitHub

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

Jump to bottom

Resnet Avg pool in fails due to reshape getting unexpected input shape #1574

Open
dgolubovicTT opened this issue Dec 12, 2024 · 0 comments
Open

Resnet Avg pool in fails due to reshape getting unexpected input shape #1574

dgolubovicTT opened this issue Dec 12, 2024 · 0 comments
Assignees
@mtopalovicTT
Milestone
[FFE - E2E] ResNe...

Comments

@dgolubovicTT
Copy link
Contributor

dgolubovicTT commented Dec 12, 2024 •
edited
Loading

This error was caught while running ResNet unique op tests.

Repro is available on branch dgolubovic/add-resnet-unique-op-tests-that-fail, and will hopefully be in main soon.

Run: pytest -svv forge/test/mlir/test_model_unique_ops.py::test_avg_pool2d_resnet

Error:

E       RuntimeError: TT_FATAL @ /localdev/dgolubovic/repos/tt-forge-fe/third_party/tt-mlir/third_party/tt-metal/src/tt-metal/ttnn/cpp/ttnn/tensor/tensor_utils.cpp:474: new_volume == old_volume
E       info:
E       Invalid arguments to reshape

I am attaching ttir and ttnnir:
AvgPool2d_resnet_ttir.txt
AvgPool2d_resnet_ttnn.txt
@dgolubovicTT dgolubovicTT added this to the [FFE - E2E] ResNet 50 milestone Dec 12, 2024
@dgolubovicTT dgolubovicTT changed the title Avg pool in ResNet fails due to reshape getting unexpected input shape Resnet Avg pool in fails due to reshape getting unexpected input shape Dec 12, 2024
@dgolubovicTT dgolubovicTT mentioned this issue Dec 12, 2024
Merged
@mtopalovicTT mtopalovicTT mentioned this issue Dec 27, 2024
Closed
@dgolubovicTT dgolubovicTT mentioned this issue Jan 23, 2025
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@mtopalovicTT mtopalovicTT

Labels
None yet
Projects
None yet
Milestone
[FFE - E2E] ResNet 50
Development

Successfully merging a pull request may close this issue.

Fixing reduction ops tenstorrent/tt-mlir
2 participants
@dgolubovicTT @mtopalovicTT