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 index 210f4bcb..615f772a 100644 --- 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: