#7823: Increase timeout and remove 1k test

tenstorrent · Jun 19, 2024 · 74b593c · 74b593c
1 parent a463452
commit 74b593c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/perf-device-models.yaml b/.github/workflows/perf-device-models.yaml
@@ -15,7 +15,7 @@ jobs:
       matrix:
         test-info: [
           {name: "GS", arch: grayskull, runs-on: ["perf-no-reset-grayskull", "self-reset"], machine-type: "bare_metal", timeout: 75},
-          {name: "N300 WH B0", arch: wormhole_b0, runs-on: ["perf-wormhole_b0", "self-reset"], machine-type: "bare_metal", timeout: 20},
+          {name: "N300 WH B0", arch: wormhole_b0, runs-on: ["perf-wormhole_b0", "self-reset"], machine-type: "bare_metal", timeout: 30},
         ]
     name: "${{ matrix.test-info.name }} device perf"
     env:

diff --git a/models/demos/falcon7b/tests/test_falcon_device_perf.py b/models/demos/falcon7b/tests/test_falcon_device_perf.py
@@ -71,7 +71,7 @@ def test_device_perf_wh_bare_metal(
 
 
 @pytest.mark.models_device_performance_bare_metal
-@pytest.mark.parametrize("seq_len, samples", [(128, 1700), (1024, 2800), (2048, 2600)])
+@pytest.mark.parametrize("seq_len, samples", [(128, 1700), (2048, 2600)])
 def test_device_perf(seq_len, samples):
     margin = 0.03
     num_iterations = 1