diff --git a/.github/workflows/_npu-single-node-test-stage.yml b/.github/workflows/_npu-single-node-test-stage.yml index abcb568bb..c0895c29a 100644 --- a/.github/workflows/_npu-single-node-test-stage.yml +++ b/.github/workflows/_npu-single-node-test-stage.yml @@ -58,6 +58,11 @@ on: description: 'GitHub token for API calls' type: string default: ${{ github.token }} + test_timeout_minutes: + required: false + type: string + default: '300' + description: 'timeout-minutes for the Run test step' env: SKIP_PR_TEST_HEALTH_CHECK: ${{ inputs.skip_pr_test_health_check }} @@ -210,7 +215,7 @@ jobs: curl -o /tmp/test.jsonl -L https://gh-proxy.test.osinfra.cn/https://raw.githubusercontent.com/openai/grade-school-math/master/grade_school_math/data/test.jsonl - name: Run test - timeout-minutes: 300 + timeout-minutes: ${{ fromJson(inputs.test_timeout_minutes) }} env: SGLANG_USE_MODELSCOPE: true HF_ENDPOINT: https://hf-mirror.com diff --git a/.github/workflows/nightly-test-npu.yml b/.github/workflows/nightly-test-npu.yml index b33d1febd..6cde2e662 100644 --- a/.github/workflows/nightly-test-npu.yml +++ b/.github/workflows/nightly-test-npu.yml @@ -278,6 +278,7 @@ jobs: install_sglang_deps: false device_type_for_deps: 'a3' run_start_metadata: ${{ needs.set-image-config.outputs.run_start_metadata }} + test_timeout_minutes: '180' # The acc-2 suite is split across 4 parallel partition jobs via the single `partitions` input. nightly-acc-2-npu-a3: diff --git a/.github/workflows/pr-test-npu.yml b/.github/workflows/pr-test-npu.yml index 70d6f5b2b..e1c0e7a86 100644 --- a/.github/workflows/pr-test-npu.yml +++ b/.github/workflows/pr-test-npu.yml @@ -394,6 +394,7 @@ jobs: image: ${{ needs.set-image-config.outputs.CANN_image_a3 }} install_sglang_deps: true device_type_for_deps: 'a3' + test_timeout_minutes: '180' pr-test-npu-finish: