extend timeout for b200 test (#14925)
Co-authored-by: Brayden Zhong <b8zhong@users.noreply.github.com>
This commit is contained in:
2
.github/workflows/pr-test.yml
vendored
2
.github/workflows/pr-test.yml
vendored
@@ -1214,7 +1214,7 @@ jobs:
|
||||
timeout-minutes: 30
|
||||
run: |
|
||||
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
|
||||
# unit-test-backend-4-gpu-gb200:
|
||||
|
||||
@@ -151,6 +151,7 @@ class Envs:
|
||||
# Test & Debug
|
||||
SGLANG_IS_IN_CI = EnvBool(False)
|
||||
SGLANG_IS_IN_CI_AMD = EnvBool(False)
|
||||
IS_BLACKWELL = EnvBool(False)
|
||||
SGLANG_SET_CPU_AFFINITY = EnvBool(False)
|
||||
SGLANG_PROFILE_WITH_STACK = EnvBool(True)
|
||||
SGLANG_PROFILE_RECORD_SHAPES = EnvBool(True)
|
||||
|
||||
@@ -30,6 +30,7 @@ from PIL import Image
|
||||
|
||||
from sglang.bench_serving import run_benchmark
|
||||
from sglang.global_config import global_config
|
||||
from sglang.srt.environ import envs
|
||||
from sglang.srt.utils import (
|
||||
get_bool_env_var,
|
||||
get_device,
|
||||
@@ -162,7 +163,7 @@ def is_in_amd_ci():
|
||||
|
||||
def is_blackwell_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):
|
||||
|
||||
Reference in New Issue
Block a user