error: 'ttnn.max_pool2d' op ttnn.max_pool2d currently only supports an input type of bfloat16. Recieved 'f32'. - Githubissues

tenstorrent / tt-mlir

Tenstorrent MLIR compiler

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

Apache License 2.0

77 stars 13 forks source link

error: 'ttnn.max_pool2d' op ttnn.max_pool2d currently only supports an input type of bfloat16. Recieved 'f32'. #1371

Open kamalrajkannan78 opened 3 days ago

kamalrajkannan78 commented 3 days ago

Description

error: 'ttnn.max_pool2d' op ttnn.max_pool2d currently only supports an input type of bfloat16. Recieved 'f32'.

Reproduce

git checkout kkannan/models_mlir_issues 
git submodule update --recursive
cmake --build build -- install_ttforge
pytest forge/test/model_demos/high_prio/cnn/pytorch/test_vovnet.py -k "not timm" -svv

Observed Behaviour

error: 'ttnn.max_pool2d' op ttnn.max_pool2d currently only supports an input type of bfloat16. Recieved 'f32'.

Logs

vovnet_issue.log

TTIR graphs