#7823: Add Falcon7b to device perf pipeline

tenstorrent · Jun 21, 2024 · 5904433 · 5904433
1 parent f64f407
commit 5904433
Show file tree

Hide file tree

Showing 9 changed files with 476 additions and 425 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: 40},
-          {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/ci/test_falcon_end_to_end_prefill.py b/models/demos/falcon7b/tests/ci/test_falcon_end_to_end_prefill.py