From 2753e51a02deacf4056b19a8e4280f683afccab0 Mon Sep 17 00:00:00 2001 From: Kartik Paigwar <132708568+kpaigwar@users.noreply.github.com> Date: Wed, 15 May 2024 14:26:35 -0400 Subject: [PATCH] #0: Update test_mamba_perf.py --- models/demos/mamba/tests/test_mamba_perf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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