diff --git a/models/demos/bert_tiny/tests/test_performance.py b/models/demos/bert_tiny/tests/test_performance.py index 4f98ed79ee7c..591740ac51bb 100644 --- a/models/demos/bert_tiny/tests/test_performance.py +++ b/models/demos/bert_tiny/tests/test_performance.py @@ -119,7 +119,7 @@ def test_perf_device_bare_metal(batch_size, expected_perf): margin = 0.03 if is_wormhole_b0(): - expected_perf = 3990.0 + expected_perf = 4114.8 else: expected_perf = 3460.0