#0: Update test_mamba_perf.py

tenstorrent · May 15, 2024 · c20298a · c20298a
1 parent 3610c23
commit c20298a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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