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

ttnn.Conv1d fails with "Statically allocated circular buffers" in SqueezeBERT model [WH] #13195

Open kkeerthana0573 opened 1 day ago

kkeerthana0573 commented 1 day ago

Describe the bug

  1. ttnn.Conv1d fails with "Statically allocated circular buffers on core range {}" in SqueezeBERT model
  2. The model has ttnn.Conv1d with input_channels and output_channels > 2560, which are skipped on main. This blocks porting the model to ttnn completely.

on WH - n300 :

Statically allocated circular buffers on core range {} grow to {} B which is beyond max L1 size of {} B

To Reproduce Steps to reproduce the behavior:

  1. Pick the test from the commit.
  2. Run pytest tests/ttnn/unit_tests/operations/test_conv1d.py::test_squeezebert_conv1d.

Expected behavior Expected to pass the test case.

Please complete the following environment information:

kkeerthana0573 commented 1 day ago

cc: @boris-drazic