[Model] Fixup tri_basic_2

(cherry picked from commit 1283b4bd3d5e7ed53f75d20b3b165bafe004beb6)
tenstorrent · Aug 2, 2024 · d17d27c · d17d27c
1 parent 9955228
commit d17d27c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pybuda/test/model_demos/high_prio/cnn/pytorch/test_tri_basic_2.py b/pybuda/test/model_demos/high_prio/cnn/pytorch/test_tri_basic_2.py
@@ -34,6 +34,9 @@ def test_tri_basic_2_sematic_segmentation_pytorch(test_device):
         compiler_cfg.balancer_op_override(
             "add_156", "t_stream_shape", (1, 1)
         )  # TM error
+        compiler_cfg.balancer_op_override(
+            "add_200", "t_stream_shape", (1, 1)
+        )  # TM error
         compiler_cfg.balancer_op_override(
             "add_214", "t_stream_shape", (1, 1)
         )  # TM error