minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

extend timeout for b200 test (#14925)

Browse Source 
Co-authored-by: Brayden Zhong <b8zhong@users.noreply.github.com>
This commit is contained in:
b8zhong
2025-12-11 15:43:53 -08:00
committed by GitHub
co-authored by Brayden Zhong
parent 5c8bd8b51b
commit 6107268fe7
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/pr-test.yml
+1 -1
View File
@@ -1214,7 +1214,7 @@ jobs:
timeout-minutes: 30 timeout-minutes: 30
run: | run: |
cd test/srt cd test/srt
python3 run_suite.py --suite per-commit-4-gpu-b200 --auto-partition-id ${{ matrix.part }} --auto-partition-size 3 --timeout-per-file 1800 IS_BLACKWELL=1 python3 run_suite.py --suite per-commit-4-gpu-b200 --auto-partition-id ${{ matrix.part }} --auto-partition-size 3 --timeout-per-file 1800
# TODO: Add gb200 tests back after the ci runner is fixed # TODO: Add gb200 tests back after the ci runner is fixed
# unit-test-backend-4-gpu-gb200: # unit-test-backend-4-gpu-gb200:
python/sglang/srt/environ.py
+1
View File
@@ -151,6 +151,7 @@ class Envs:
# Test & Debug # Test & Debug
SGLANG_IS_IN_CI = EnvBool(False) SGLANG_IS_IN_CI = EnvBool(False)
SGLANG_IS_IN_CI_AMD = EnvBool(False) SGLANG_IS_IN_CI_AMD = EnvBool(False)
IS_BLACKWELL = EnvBool(False)
SGLANG_SET_CPU_AFFINITY = EnvBool(False) SGLANG_SET_CPU_AFFINITY = EnvBool(False)
SGLANG_PROFILE_WITH_STACK = EnvBool(True) SGLANG_PROFILE_WITH_STACK = EnvBool(True)
SGLANG_PROFILE_RECORD_SHAPES = EnvBool(True) SGLANG_PROFILE_RECORD_SHAPES = EnvBool(True)
python/sglang/test/test_utils.py
+2 -1
View File
@@ -30,6 +30,7 @@ from PIL import Image
from sglang.bench_serving import run_benchmark from sglang.bench_serving import run_benchmark
from sglang.global_config import global_config from sglang.global_config import global_config
from sglang.srt.environ import envs
from sglang.srt.utils import ( from sglang.srt.utils import (
get_bool_env_var, get_bool_env_var,
get_device, get_device,
@@ -162,7 +163,7 @@ def is_in_amd_ci():
def is_blackwell_system(): def is_blackwell_system():
"""Return whether it is running on a Blackwell (B200) system.""" """Return whether it is running on a Blackwell (B200) system."""
return get_bool_env_var("IS_BLACKWELL") return envs.IS_BLACKWELL.get()
def _use_cached_default_models(model_repo: str): def _use_cached_default_models(model_repo: str):