Regression: Fix Whisper Medium on grayskull e300

(cherry picked from commit af37bdb86e3ffe0ea5c5dc5ffcf02a0b71f11d3e)
tenstorrent · Apr 4, 2024 · 2f4a8ee · 2f4a8ee
1 parent 8766d56
commit 2f4a8ee
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pybuda/test/model_demos/high_prio/nlp/pytorch/test_whisper_1.py b/pybuda/test/model_demos/high_prio/nlp/pytorch/test_whisper_1.py
@@ -120,6 +120,8 @@ def test_whisper_enc_dec(test_device, variant):
         compiler_cfg.enable_auto_fusing = False
         if "large" in variant:
             compiler_cfg.place_on_new_epoch("matmul_2805")
+        if "medium" in variant:
+            compiler_cfg.place_on_new_epoch("matmul_3295")
         if variant == "openai/whisper-base":
             compiler_cfg.amp_level = 1
         else: