Npu single node test timeout config (#36092)

.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

.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:

.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: