-
Notifications
You must be signed in to change notification settings - Fork 254
Validate official vLLM Router with srt-slurm #2543
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Open
cquil11
wants to merge
20
commits into
main
Choose a base branch
from
agent/native-router-integration
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
Changes from all commits
Commits
Show all changes
20 commits
Select commit
Hold shift + click to select a range
f19290d
feat: exercise native model routers through srt-slurm
cquil11 5c98e20
docs: register native router validation sweep
cquil11 53634d7
fix: verify vLLM Router through package metadata
cquil11 37243c4
fix native router validation recipes
cquil11 d50adec
fix: enable MNNVL for B200 vLLM router workers
cquil11 0bd3585
fix: use proven B200 vLLM communicator path
cquil11 05b4761
fix: use CUDA-aware UCX for SGL Router P/D
cquil11 d142e6e
chore: pin router startup-timeout support
cquil11 be0da91
test: move SGL Router P/D validation to H200
cquil11 62078c4
fix: enable native routers on H200 launcher
cquil11 31a4444
fix: pin native routers after SGL P/D warmup fix
cquil11 35e586b
Validate native router jobs and pin PD fix
cquil11 32671d2
Harden native router validation
cquil11 6efbcfc
Merge remote-tracking branch 'origin/main' into agent/native-router-i…
cquil11 094b1ac
ci: recognize native router runtimes in sweep classifier
cquil11 a983e78
ci: harden native router sweep handling
cquil11 f58cd40
ci: pin latest validated srt-slurm router head
cquil11 20a1cd7
use semantic native router frontend names
cquil11 95cc2cc
test official vLLM Router smoke points
cquil11 406a6e4
Merge remote-tracking branch 'origin/main' into agent/native-router-i…
cquil11 File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
5 changes: 5 additions & 0 deletions
5
benchmarks/multi_node/srt-slurm-recipes/configs/native-sgl-router-deps.sh
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,5 @@ | ||
| #!/usr/bin/env bash | ||
| set -euo pipefail | ||
|
|
||
| python3 -m pip install --no-cache-dir --upgrade "sglang-router==0.3.2" | ||
| python3 -c 'import sglang_router; print(sglang_router.__file__)' |
55 changes: 55 additions & 0 deletions
55
benchmarks/multi_node/srt-slurm-recipes/configs/native-sgl-router-pd-fix-deps.sh
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,55 @@ | ||
| #!/usr/bin/env bash | ||
| set -euo pipefail | ||
|
|
||
| # Temporary exact-source pin for the upstream SGLang Model Gateway one-token | ||
| # P/D response fix. The shared wheel cache and flock make the source build | ||
| # happen once even though srt-slurm runs the recipe setup on every role. | ||
| SGLANG_REPOSITORY="https://github.com/cquil11/sglang.git" | ||
| SGLANG_REF="fix/pd-one-token-prefill-response" | ||
| SGLANG_COMMIT="9072e0dc5aaf1962b4e2c6f1a51094356b5d3324" | ||
| WHEEL_CACHE_ROOT="/router_wheels/sglang-router-${SGLANG_COMMIT}" | ||
| LOCK_FILE="${WHEEL_CACHE_ROOT}.lock" | ||
|
|
||
| mkdir -p "${WHEEL_CACHE_ROOT}" | ||
| exec 9>"${LOCK_FILE}" | ||
| flock -w 3600 9 | ||
|
|
||
| shopt -s nullglob | ||
| router_wheels=("${WHEEL_CACHE_ROOT}"/sglang_router-*.whl) | ||
| if (( ${#router_wheels[@]} == 0 )); then | ||
| build_dir=$(mktemp -d /tmp/sglang-router-build.XXXXXX) | ||
| trap 'rm -rf "${build_dir}"' EXIT | ||
|
|
||
| git clone --filter=blob:none --no-checkout --single-branch \ | ||
| --branch "${SGLANG_REF}" "${SGLANG_REPOSITORY}" "${build_dir}" | ||
| git -C "${build_dir}" sparse-checkout init --cone | ||
| git -C "${build_dir}" sparse-checkout set sgl-model-gateway | ||
| git -C "${build_dir}" checkout --detach "${SGLANG_COMMIT}" | ||
| if [[ "$(git -C "${build_dir}" rev-parse HEAD)" != "${SGLANG_COMMIT}" ]]; then | ||
| echo "ERROR: SGLang checkout does not match ${SGLANG_COMMIT}" >&2 | ||
| exit 1 | ||
| fi | ||
|
|
||
| python3 -m pip install --break-system-packages --no-cache-dir --upgrade \ | ||
| "maturin>=1.0,<2.0" | ||
| ( | ||
| cd "${build_dir}/sgl-model-gateway/bindings/python" | ||
| CARGO_BUILD_JOBS="${CARGO_BUILD_JOBS:-16}" \ | ||
| maturin build --release --out "${WHEEL_CACHE_ROOT}" | ||
| ) | ||
| router_wheels=("${WHEEL_CACHE_ROOT}"/sglang_router-*.whl) | ||
| fi | ||
|
|
||
| if (( ${#router_wheels[@]} != 1 )); then | ||
| echo "ERROR: expected exactly one cached SGLang Router wheel, found ${#router_wheels[@]}" >&2 | ||
| exit 1 | ||
| fi | ||
|
|
||
| # Install the released package first to resolve its Python dependencies, then | ||
| # replace only the native package with the wheel built from the exact fix SHA. | ||
| python3 -m pip install --break-system-packages --no-cache-dir --upgrade \ | ||
| "sglang-router==0.3.2" | ||
| python3 -m pip install --break-system-packages --no-cache-dir \ | ||
| --force-reinstall --no-deps "${router_wheels[0]}" | ||
| python3 -c 'import sglang_router; print(sglang_router.__file__)' | ||
| echo "Installed SGLang Router from ${SGLANG_COMMIT}" |
11 changes: 11 additions & 0 deletions
11
benchmarks/multi_node/srt-slurm-recipes/configs/native-vllm-router-deps.sh
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| #!/usr/bin/env bash | ||
| set -euo pipefail | ||
|
|
||
| python3 -m pip install --no-cache-dir --upgrade "vllm-router==0.1.15" | ||
| command -v vllm-router >/dev/null | ||
| python3 - <<'PY' | ||
| from importlib.metadata import version | ||
|
|
||
| assert version("vllm-router") == "0.1.15" | ||
| print(f"vllm-router {version('vllm-router')}") | ||
| PY |
96 changes: 96 additions & 0 deletions
96
.../srt-slurm-recipes/sglang/deepseek-v4/agentic/agg-b200-2xdep8-mtp-sgl-router-agentic.yaml
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,96 @@ | ||
| name: "dsv4-b200-sgl-router-agg-2xdep8-mtp-agentic" | ||
|
|
||
| model: | ||
| path: "deepseek-v4-pro" | ||
| container: "lmsysorg/sglang:v0.5.16-cu130" | ||
| precision: "fp4" | ||
|
|
||
| setup_script: native-sgl-router-deps.sh | ||
|
|
||
| slurm: | ||
| time_limit: "4:00:00" | ||
|
|
||
| health_check: | ||
| max_attempts: 720 | ||
| interval_seconds: 10 | ||
|
|
||
| resources: | ||
| gpu_type: "b200" | ||
| gpus_per_node: 8 | ||
| agg_nodes: 2 | ||
| agg_workers: 2 | ||
| gpus_per_agg: 8 | ||
|
|
||
| infra: | ||
| etcd_nats_dedicated_node: false | ||
|
|
||
| frontend: | ||
| type: sglang | ||
| enable_multiple_frontends: false | ||
| env: | ||
| PIP_BREAK_SYSTEM_PACKAGES: "1" | ||
|
|
||
| backend: | ||
| type: sglang | ||
| aggregated_environment: | ||
| PIP_BREAK_SYSTEM_PACKAGES: "1" | ||
| PYTHONUNBUFFERED: "1" | ||
| SGLANG_DEFAULT_THINKING: "1" | ||
| SGLANG_DSV4_REASONING_EFFORT: "high" | ||
| SGLANG_SIMULATE_ACC_LEN: "2.99" | ||
| SGLANG_SIMULATE_ACC_METHOD: "match-expected" | ||
| SGLANG_SIMULATE_ACC_TOKEN_MODE: "real-draft-token" | ||
| SGLANG_JIT_DEEPGEMM_PRECOMPILE: "1" | ||
| SGLANG_OPT_SWA_SPLIT_LEAF_ON_INSERT: "1" | ||
| SGLANG_OPT_UNIFIED_CACHE_FREE_OUT_OF_WINDOW_SLOTS: "1" | ||
| SGLANG_OPT_USE_CUSTOM_ALL_REDUCE_V2: "1" | ||
| SGLANG_OPT_USE_JIT_INDEXER_METADATA: "1" | ||
| SGLANG_OPT_USE_JIT_NORM: "1" | ||
| SGLANG_OPT_USE_TOPK_V2: "True" | ||
| NCCL_CUMEM_ENABLE: "1" | ||
| NCCL_NVLS_ENABLE: "1" | ||
| sglang_config: | ||
| aggregated: | ||
| served-model-name: "deepseek-ai/DeepSeek-V4-Pro" | ||
| enable-metrics: true | ||
| enable-cache-report: true | ||
| trust-remote-code: true | ||
| tool-call-parser: "deepseekv4" | ||
| stream-interval: 10 | ||
| watchdog-timeout: 1800 | ||
| mem-fraction-static: 0.90 | ||
| chunked-prefill-size: 8192 | ||
| moe-runner-backend: "flashinfer_mxfp4" | ||
| disable-flashinfer-autotune: true | ||
| max-running-requests: 256 | ||
| cuda-graph-max-bs: 256 | ||
| scheduler-recv-interval: 30 | ||
| dp-size: 8 | ||
| tp-size: 8 | ||
| ep-size: 8 | ||
| enable-dp-attention: true | ||
| enable-dp-lm-head: true | ||
| speculative-algorithm: "EAGLE" | ||
| speculative-num-steps: 3 | ||
| speculative-eagle-topk: 1 | ||
| speculative-num-draft-tokens: 4 | ||
|
|
||
| sbatch_directives: | ||
| cpus-per-task: "144" | ||
| mem: "0" | ||
|
|
||
| srun_options: | ||
| container-remap-root: "" | ||
|
|
||
| benchmark: | ||
| type: custom | ||
| command: bash /infmax-workspace/benchmarks/multi_node/agentic_srt.sh | ||
| env: | ||
| INFMAX_CONTAINER_WORKSPACE: "/infmax-workspace" | ||
| RESULT_DIR: "/logs/agentic" | ||
| PORT: "8000" | ||
| IS_MULTINODE: "true" | ||
| AIPERF_REQUIRED_SERVER_METRIC_PREFIX: "sglang:" | ||
| AIPERF_DATASET_MMAP_CACHE_DIR: "/aiperf_mmap_cache" | ||
| HF_HUB_CACHE: "/hf_hub_cache" | ||
| WEKA_LOADER_OVERRIDE: "semianalysis_cc_traces_weka_062126" | ||
152 changes: 152 additions & 0 deletions
152
...srt-slurm-recipes/sglang/glm5.2/agentic/disagg-h200-2p2d-pcp8-tp8-dp8-mtp-sgl-router.yaml
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,152 @@ | ||
| name: "glm5.2-h200-sgl-router-disagg-2p2d-mtp-agentic" | ||
|
|
||
| # Native SGL Router validation of the established GLM-5.2 H200 Mooncake | ||
| # topology. This deliberately changes only the request plane: model, worker | ||
| # layout, KV transport, MTP, and AgentX behavior match the Dynamo baseline. | ||
|
|
||
| model: | ||
| path: "hf:zai-org/GLM-5.2-FP8" | ||
| container: "lmsysorg/sglang:v0.5.16-cu130" | ||
| precision: "fp8" | ||
|
|
||
| identity: | ||
| model: | ||
| repo: "zai-org/GLM-5.2-FP8" | ||
| revision: "ba978f7d347eaf65d22f1a86833408afdb953541" | ||
| container: | ||
| image: "lmsysorg/sglang:v0.5.16-cu130" | ||
| frameworks: | ||
| sglang: "0.5.16" | ||
|
|
||
| setup_script: native-sgl-router-pd-fix-deps.sh | ||
|
|
||
| slurm: | ||
| time_limit: "8:00:00" | ||
|
|
||
| health_check: | ||
| max_attempts: 720 | ||
| interval_seconds: 10 | ||
|
|
||
| resources: | ||
| gpu_type: h200 | ||
| gpus_per_node: 8 | ||
| prefill_nodes: 2 | ||
| decode_nodes: 2 | ||
| prefill_workers: 2 | ||
| decode_workers: 2 | ||
| gpus_per_prefill: 8 | ||
| gpus_per_decode: 8 | ||
|
|
||
| infra: | ||
| etcd_nats_dedicated_node: false | ||
|
|
||
| frontend: | ||
| type: sglang | ||
| enable_multiple_frontends: false | ||
| env: | ||
| PIP_BREAK_SYSTEM_PACKAGES: "1" | ||
|
|
||
| backend: | ||
| type: sglang | ||
| prefill_environment: | ||
| HF_HOME: "/hf_hub_cache" | ||
| HF_HUB_CACHE: "/hf_hub_cache" | ||
| PIP_BREAK_SYSTEM_PACKAGES: "1" | ||
| PYTHONUNBUFFERED: "1" | ||
| SGLANG_DISAGGREGATION_WAITING_TIMEOUT: "900" | ||
| SGLANG_MOONCAKE_CUSTOM_MEM_POOL: "True" | ||
| SGLANG_OPT_USE_TOPK_V2: "1" | ||
| SGLANG_TIMEOUT_KEEP_ALIVE: "900" | ||
| decode_environment: | ||
| HF_HOME: "/hf_hub_cache" | ||
| HF_HUB_CACHE: "/hf_hub_cache" | ||
| PIP_BREAK_SYSTEM_PACKAGES: "1" | ||
| PYTHONUNBUFFERED: "1" | ||
| SGLANG_DISAGGREGATION_WAITING_TIMEOUT: "900" | ||
| SGLANG_MOONCAKE_CUSTOM_MEM_POOL: "True" | ||
| SGLANG_OPT_USE_TOPK_V2: "1" | ||
| SGLANG_TIMEOUT_KEEP_ALIVE: "900" | ||
| SGLANG_SIMULATE_ACC_LEN: "2.99" | ||
| SGLANG_SIMULATE_ACC_METHOD: "match-expected" | ||
| SGLANG_SIMULATE_ACC_TOKEN_MODE: "real-draft-token" | ||
|
|
||
| sglang_config: | ||
| prefill: | ||
| host: 0.0.0.0 | ||
| model-path: /model/ | ||
| served-model-name: zai-org/GLM-5.2-FP8 | ||
| trust-remote-code: true | ||
| tool-call-parser: glm47 | ||
| reasoning-parser: glm45 | ||
| tp-size: 8 | ||
| ep-size: 1 | ||
| attn-cp-size: 8 | ||
| enable-prefill-cp: true | ||
| cp-strategy: interleave | ||
| enable-dsa-cache-layer-split: true | ||
| disaggregation-transfer-backend: mooncake | ||
| disaggregation-mode: prefill | ||
| kv-cache-dtype: fp8_e4m3 | ||
| context-length: 1048576 | ||
| max-total-tokens: 1048576 | ||
| chunked-prefill-size: 32768 | ||
| mem-fraction-static: 0.85 | ||
| max-running-requests: 32 | ||
| speculative-algorithm: EAGLE | ||
| speculative-num-steps: 3 | ||
| speculative-eagle-topk: 1 | ||
| speculative-num-draft-tokens: 4 | ||
| watchdog-timeout: 86400 | ||
| stream-interval: 60 | ||
| enable-metrics: true | ||
| enable-cache-report: true | ||
|
|
||
| decode: | ||
| host: 0.0.0.0 | ||
| model-path: /model/ | ||
| served-model-name: zai-org/GLM-5.2-FP8 | ||
| trust-remote-code: true | ||
| tool-call-parser: glm47 | ||
| reasoning-parser: glm45 | ||
| tp-size: 8 | ||
| dp-size: 8 | ||
| ep-size: 1 | ||
| enable-dp-attention: true | ||
| disaggregation-transfer-backend: mooncake | ||
| disaggregation-mode: decode | ||
| kv-cache-dtype: fp8_e4m3 | ||
| dsa-decode-backend: flashmla_kv | ||
| context-length: 1048576 | ||
| max-total-tokens: 1048576 | ||
| mem-fraction-static: 0.85 | ||
| max-running-requests: 200 | ||
| page-size: 64 | ||
| disable-radix-cache: true | ||
| speculative-algorithm: EAGLE | ||
| speculative-num-steps: 3 | ||
| speculative-eagle-topk: 1 | ||
| speculative-num-draft-tokens: 4 | ||
| watchdog-timeout: 86400 | ||
| stream-interval: 60 | ||
| enable-metrics: true | ||
| enable-cache-report: true | ||
|
|
||
| sbatch_directives: | ||
| mem: "0" | ||
|
|
||
| srun_options: | ||
| mem: "0" | ||
| container-remap-root: "" | ||
|
|
||
| benchmark: | ||
| type: custom | ||
| command: bash /infmax-workspace/benchmarks/multi_node/agentic_srt.sh | ||
| env: | ||
| INFMAX_CONTAINER_WORKSPACE: /infmax-workspace | ||
| RESULT_DIR: /logs/agentic | ||
| PORT: "8000" | ||
| IS_MULTINODE: "true" | ||
| AIPERF_REQUIRED_SERVER_METRIC_PREFIX: "sglang:" | ||
| AIPERF_DATASET_MMAP_CACHE_DIR: "/aiperf_mmap_cache" | ||
| HF_HUB_CACHE: "/hf_hub_cache" | ||
| WEKA_LOADER_OVERRIDE: "semianalysis_cc_traces_weka_062126" |
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.