extend timeout for b200 test (#14925)

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
parent 5c8bd8b51b
commit 6107268fe7
3 changed files with 4 additions and 2 deletions

View File

@@ -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)

View File

@@ -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):