From 2f4a8ee76cb134c293029fccf478d9acf97e5428 Mon Sep 17 00:00:00 2001 From: Allan Rui Date: Wed, 3 Apr 2024 16:26:48 +0000 Subject: [PATCH] Regression: Fix Whisper Medium on grayskull e300 (cherry picked from commit af37bdb86e3ffe0ea5c5dc5ffcf02a0b71f11d3e) --- pybuda/test/model_demos/high_prio/nlp/pytorch/test_whisper_1.py | 2 ++ 1 file changed, 2 insertions(+) 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: