Attribute 'shape' error in 'ttir.reshape' op in adaptive_avg_pool_2d

tenstorrent / tt-forge-fe

The TT-Forge FE is a graph compiler designed to optimize and transform computational graphs for deep learning models, enhancing their performance and efficiency.

https://docs.tenstorrent.com/tt-forge-fe/

Apache License 2.0

19 stars 4 forks source link

Attribute 'shape' error in 'ttir.reshape' op in adaptive_avg_pool_2d #559

Closed kamalrajkannan78 closed 4 days ago

kamalrajkannan78 commented 5 days ago

Description

adaptive_avg_pool_2d faced loc("avg_pool2d_16.dc.reshape.0"("forward":4294967295:78)): error: 'ttir.reshape' op requires attribute 'shape'

Reproduce

git checkout kkannan/adaptive_avg_pool
git submodule update --recursive
pytest forge/test/mlir/test_ops.py -k "issue" -svv

Observed Behaviour

loc("avg_pool2d_16.dc.reshape.0"("forward":4294967295:78)): error: 'ttir.reshape' op requires attribute 'shape'

Logs

reshape_issue.log
reshape_fix.log

kamalrajkannan78 commented 5 days ago

FYI @dgolubovicTT