#7525: Re-enable perf tests

tenstorrent · Mar 6, 2025 · 9107c02 · 9107c02
1 parent e4dc25d
commit 9107c02
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/models/demos/metal_BERT_large_11/tests/test_perf_bert11.py b/models/demos/metal_BERT_large_11/tests/test_perf_bert11.py
@@ -151,7 +151,7 @@ def run_perf_bert11(
     logger.info(f"bert11 compile time: {compile_time}")
 
 
-@pytest.mark.skipif(is_wormhole_b0() or is_blackhole(), reason="May hang, need to check functional tests")
+@pytest.mark.skipif(is_blackhole(), reason="Not functional on BH")
 @run_for_wormhole_b0(reason_str="WH specific batch size")
 @pytest.mark.models_performance_bare_metal
 @pytest.mark.parametrize(

diff --git a/models/demos/metal_BERT_large_11/tests/test_perf_device_bert.py b/models/demos/metal_BERT_large_11/tests/test_perf_device_bert.py
@@ -44,7 +44,6 @@ def test_perf_device_bare_metal(batch_size, test, expected_perf):
     run_bert_perf(batch_size, test, expected_perf)
 
 
-@pytest.mark.skip("#7525: Hangs non-deterministically on device perf, likely same issue as demo model")
 @skip_for_grayskull("Incorrect device metrics for grayskull")
 @pytest.mark.models_device_performance_bare_metal
 @pytest.mark.parametrize(