"Runtime Error: Reshape Operation Fails Due to Mismatched Tensor Volume" - Githubissues

tenstorrent / tt-mlir

Tenstorrent MLIR compiler

https://tenstorrent.github.io/tt-mlir/

Apache License 2.0

77 stars 13 forks source link

"Runtime Error: Reshape Operation Fails Due to Mismatched Tensor Volume" #1377

Open kamalrajkannan78 opened 3 days ago

kamalrajkannan78 commented 3 days ago

Description

efficientnet torchvision variants faces Invalid arguments to reshape runtime error

Reproduce

git checkout kkannan/model_issues_mlir_nov26
git submodule update --recursive
cmake --build build -- install_ttforge
pytest forge/test/model_demos/high_prio/cnn/pytorch/test_efficientnet.py -k "torchvision" -svv

Observed Behaviour

E       RuntimeError: TT_FATAL @ /proj_sw/user_dev/kkannan/tt-forge-fe/third_party/tt-mlir/third_party/tt-metal/src/tt-metal/ttnn/cpp/ttnn/tensor/tensor_utils.cpp:483: new_volume == old_volume
E       info:
E       Invalid arguments to reshape

Logs

efficientnet_torchvision_variants_issue.log

TTIR graph