From 74b593c5e72c195aa8538455cbfc641a98fd727c Mon Sep 17 00:00:00 2001 From: Sofija Jovic Date: Wed, 19 Jun 2024 07:23:22 +0000 Subject: [PATCH] #7823: Increase timeout and remove 1k test --- .github/workflows/perf-device-models.yaml | 2 +- models/demos/falcon7b/tests/test_falcon_device_perf.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/perf-device-models.yaml b/.github/workflows/perf-device-models.yaml index d794f3760601..5ebcaa996edd 100644 --- 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 index 6d6144bc1c57..d3185cf04ac2 100644 --- 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