feat: add cuda core dump CI warpper (#18909)

This commit is contained in:
Liangsheng Yin
2026-02-17 14:49:26 -08:00
committed by GitHub
parent 9a7d6be567
commit 83a475e8d7
7 changed files with 272 additions and 2 deletions
+7
View File
@@ -176,6 +176,8 @@ class Envs:
# SGLang CI
SGLANG_IS_IN_CI = EnvBool(False)
SGLANG_IS_IN_CI_AMD = EnvBool(False)
SGLANG_CUDA_COREDUMP = EnvBool(False)
SGLANG_CUDA_COREDUMP_DIR = EnvStr("/tmp/sglang_cuda_coredumps")
SGLANG_TEST_MAX_RETRY = EnvInt(None)
# Constrained Decoding (Grammar)
@@ -575,6 +577,11 @@ _warn_deprecated_env_to_cli_flag(
"Please use '--prefill-delayer-token-usage-low-watermark' instead.",
)
# Import cuda_coredump to trigger auto-injection of CUDA env vars
# when SGLANG_CUDA_COREDUMP=1. Best-effort; for strict guarantees,
# set CUDA_* env vars in the shell before launching Python.
import sglang.srt.debug_utils.cuda_coredump # noqa: F401, E402
def example_with_exit_stack():
# Use this style of context manager in unit test