search

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.
Apache License 2.0
415 stars 53 forks source link

MobilenetV2 conv fails with unsupported bias shape error #12227

Open keerthana-r-mcw opened 3 weeks ago

keerthana-r-mcw commented 3 weeks ago

Describe the bug Facing unsupported bias shape error in conv op of mobilenetv2.

RuntimeError: TT_FATAL @ ../ttnn/cpp/ttnn/operations/matmul/device/matmul_op.cpp:977: bias_shape.without_padding()[-1] == b_shape[-1]
E       info:
E       Unsupported bias shape: last dimension not equal to second input's last dimension.

Steps to reproduce the behavior:

  1. Checkout keerthanar/mobilenetv2_unittest_cases
  2. Run pytest tests/ttnn/unit_tests/operations/test_new_conv2d.py::test_conv_mobilenetv2_bias_issue

Please complete the following environment information:

keerthana-r-mcw commented 3 weeks ago

https://github.com/tenstorrent/tt-metal/issues/11998