diff --git a/models/demos/mamba/tests/test_mamba_perf.py b/models/demos/mamba/tests/test_mamba_perf.py index b7dfe365c44..cae6044e7d3 100644 --- a/models/demos/mamba/tests/test_mamba_perf.py +++ b/models/demos/mamba/tests/test_mamba_perf.py @@ -23,7 +23,7 @@ @pytest.mark.models_performance_bare_metal @pytest.mark.parametrize( "batch, iterations, expected_compile_time, expected_inference_time", - ((32, 10, 15.0, 0.75),), # Issue 7816 Compile time + ((32, 10, 12.5, 0.40),), # Issue 7816 Compile time ) def test_mamba_e2e_perf( device, batch, iterations, expected_compile_time, expected_inference_time, use_program_cache, reset_seeds