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

[meenakshiramanathan1]

Describe the bug
ttnn.max_pool2d op throws an assertion that only currently bfloat16 input is supported
error: 'ttnn.max_pool2d' op ttnn.max_pool2d currently only supports an input type of bfloat16. Recieved 'f32'.
This issue was observed in many models and changing the model and inputs to bfloat16 is not supported in forge currently. Hence float32 input support is required for ttnn.max_pool2d op.

To Reproduce
Steps to reproduce the behavior:
git checkout mramanathan/maxpool2d_fp32
git submodule update --recursive

pytest tests/ttnn/unit_tests/operations/test_maxpool2d.py::test_run_max_pool_fp32 -svv

Observed behaviour
E RuntimeError: TT_FATAL @ ../ttnn/cpp/ttnn/operations/pool/maxpool/device/max_pool2d_device_op.cpp:25: input.get_dtype() == DataType::BFLOAT16 E info: E Only BFLOAT16 supported for now

Logs
maxpool2d_metal_fp32.log