Skip to content

Commit 01acaba

Browse files
config: add GB200 DSpark AgentX recipes / 添加 GB200 DSpark AgentX 配置 (#3416)
* config: add GB200 DSpark AgentX recipes 新增 GB200 DSpark AgentX 配置,包含 C32 和 C64。 * chore: link GB200 DSpark changelog 为 GB200 DSpark 变更记录添加 PR 链接。 * config: include DSpark throughput runs 为 GB200 DSpark AgentX 配置添加吞吐量测试。 * config: refresh GB200 DSpark recipes 适配当前目录和启动脚本,并在 TP8 C1/C4/C8 使用 DSpark K6。 * chore: refresh PR #3416 for sweep reuse [skip-sweep] --------- Co-authored-by: Ankur-singh <ankusingh@nvidia.com>
1 parent 6f0d115 commit 01acaba

13 files changed

Lines changed: 2351 additions & 0 deletions
Lines changed: 135 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,135 @@
1+
schema: 2
2+
name: dsv4-vllm-agg-gb200-tp8-c1-dspark6-agentic
3+
identity:
4+
model:
5+
revision: 72e1d3230f6c080a530b0a1d46f8eb4602340597
6+
frameworks:
7+
dynamo: 1.3.0.dev20260720
8+
dynamo:
9+
install: true
10+
source:
11+
wheel: 1.3.0.dev20260720
12+
environment:
13+
PYTHONHASHSEED: '0'
14+
WITH_NVIDIA_PEERMEM: '0'
15+
EVAL_MAX_MODEL_LEN: '1048576'
16+
DYNAMO_WHEEL_DIRS: /srtctl-wheels
17+
ETCD_LEASE_TTL: '7200'
18+
slurm:
19+
time_limit: '2:00:00'
20+
health_check:
21+
max_attempts: 540
22+
interval_seconds: 10
23+
resources:
24+
gpu_type: gb200
25+
gpus_per_node: 4
26+
services:
27+
- name: etcd
28+
type: etcd
29+
placement:
30+
node: infra
31+
- name: nats
32+
type: nats
33+
placement:
34+
node: infra
35+
options:
36+
max_payload_mb: 32
37+
- name: mooncake-master
38+
type: mooncake-master
39+
options:
40+
store_config:
41+
metadata_server: P2PHANDSHAKE
42+
global_segment_size: 150GB
43+
local_buffer_size: 4GB
44+
protocol: rdma
45+
device_name: "@fabric.rdma-devices"
46+
mode: embedded
47+
enable_offload: false
48+
frontend:
49+
type: dynamo
50+
enable_multiple_frontends: false
51+
args:
52+
router-mode: kv
53+
router-temperature: 0
54+
router-queue-threshold: 65536
55+
active-decode-blocks-threshold: None
56+
active-prefill-tokens-threshold: None
57+
active-prefill-tokens-threshold-frac: None
58+
tokenizer: fastokens
59+
engine:
60+
type: vllm
61+
connector: null
62+
roles:
63+
agg:
64+
nodes: 2
65+
workers: 1
66+
gpus: 8
67+
env:
68+
HF_HUB_CACHE: /hf_hub_cache
69+
HUGGINGFACE_HUB_CACHE: /hf_hub_cache
70+
TRANSFORMERS_CACHE: /hf_hub_cache
71+
VLLM_ENGINE_READY_TIMEOUT_S: '3600'
72+
VLLM_RPC_TIMEOUT: '600000'
73+
VLLM_LOG_STATS_INTERVAL: '1'
74+
VLLM_V2_WARMUP_MAX_NUM_SEQS: '16'
75+
TILELANG_CLEANUP_TEMP_FILES: '1'
76+
VLLM_USE_NCCL_SYMM_MEM: '0'
77+
TORCH_SYMMMEM: NVSHMEM
78+
NCCL_CUMEM_ENABLE: '1'
79+
NCCL_MNNVL_ENABLE: '1'
80+
NCCL_NVLS_ENABLE: '1'
81+
VLLM_SERVER_DEV_MODE: '1'
82+
VLLM_USE_V2_MODEL_RUNNER: '1'
83+
VLLM_USE_RUST_FRONTEND: '1'
84+
VLLM_ALLREDUCE_USE_FLASHINFER: '1'
85+
VLLM_FLASHINFER_ALLREDUCE_BACKEND: auto
86+
VLLM_MOONCAKE_LOAD_RECV_THREADS: '20'
87+
VLLM_PREFIX_CACHE_RETENTION_INTERVAL: '32768'
88+
VLLM_SPARSE_INDEXER_MAX_LOGITS_MB: '1024'
89+
VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS: '1800'
90+
UCX_MEMTYPE_CACHE: n
91+
UCX_NET_DEVICES: "@fabric.rdma-device-ports"
92+
UCX_TLS: rc,cuda_copy
93+
NCCL_IB_HCA: "@fabric.rdma-devices"
94+
NCCL_P2P_LEVEL: NVL
95+
MC_ENABLE_DEST_DEVICE_AFFINITY: '1'
96+
MC_STORE_CLIENT_METRIC: '1'
97+
MC_STORE_CLIENT_METRIC_INTERVAL: '5'
98+
MC_TE_METRIC: '0'
99+
DG_JIT_CACHE_DIR: /tmp/dg-cache-dsv4-gb200-dspark-agg-{job_id}
100+
VLLM_USE_BREAKABLE_CUDAGRAPH: '0'
101+
args:
102+
served-model-name: deepseek-ai/DeepSeek-V4-Pro-0813
103+
kv-cache-dtype: fp8
104+
tensor-parallel-size: 8
105+
pipeline-parallel-size: 1
106+
disable-custom-all-reduce: true
107+
attention-config: '{"backend":"FLASHINFER_MLA_SPARSE_DSV4","use_prefill_query_quantization":true,"indexer_kv_dtype":"mxfp4"}'
108+
max-model-len: 1048576
109+
max-num-seqs: 16
110+
max-num-batched-tokens: 8192
111+
trust-remote-code: true
112+
no-enable-flashinfer-autotune: true
113+
block-size: 256
114+
cudagraph-metrics: true
115+
compilation-config: '{"cudagraph_mode":"FULL_DECODE_ONLY","cudagraph_capture_sizes":[7,14,21,28,35,42,49,56,63,70,77,84,91,98,105,112],"mode":0}'
116+
speculative-config: '{"method":"dspark","num_speculative_tokens":6,"draft_sample_method":"probabilistic","rejection_sample_method":"block"}'
117+
gpu-memory-utilization: 0.94
118+
stream-interval: 10
119+
no-disable-hybrid-kv-cache-manager: true
120+
tokenizer-mode: deepseek_v4
121+
sbatch_directives:
122+
cpus-per-task: '144'
123+
mem: '0'
124+
srun_options:
125+
container-remap-root: ''
126+
cpu-bind: none
127+
benchmark:
128+
type: custom
129+
command: bash /infmax-workspace/benchmarks/srt_agentic.sh
130+
env:
131+
INFMAX_CONTAINER_WORKSPACE: /infmax-workspace
132+
PORT: '8000'
133+
AIPERF_USE_DYNAMO_CONV_AWARE_ROUTING: '0'
134+
WEKA_LOADER_OVERRIDE: semianalysis_cc_traces_weka_062126
135+
AIPERF_DATASET_MMAP_CACHE_ENABLED: 'false'
Lines changed: 135 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,135 @@
1+
schema: 2
2+
name: dsv4-vllm-agg-gb200-tp8-c4-dspark6-agentic
3+
identity:
4+
model:
5+
revision: 72e1d3230f6c080a530b0a1d46f8eb4602340597
6+
frameworks:
7+
dynamo: 1.3.0.dev20260720
8+
dynamo:
9+
install: true
10+
source:
11+
wheel: 1.3.0.dev20260720
12+
environment:
13+
PYTHONHASHSEED: '0'
14+
WITH_NVIDIA_PEERMEM: '0'
15+
EVAL_MAX_MODEL_LEN: '1048576'
16+
DYNAMO_WHEEL_DIRS: /srtctl-wheels
17+
ETCD_LEASE_TTL: '7200'
18+
slurm:
19+
time_limit: '2:00:00'
20+
health_check:
21+
max_attempts: 540
22+
interval_seconds: 10
23+
resources:
24+
gpu_type: gb200
25+
gpus_per_node: 4
26+
services:
27+
- name: etcd
28+
type: etcd
29+
placement:
30+
node: infra
31+
- name: nats
32+
type: nats
33+
placement:
34+
node: infra
35+
options:
36+
max_payload_mb: 32
37+
- name: mooncake-master
38+
type: mooncake-master
39+
options:
40+
store_config:
41+
metadata_server: P2PHANDSHAKE
42+
global_segment_size: 150GB
43+
local_buffer_size: 4GB
44+
protocol: rdma
45+
device_name: "@fabric.rdma-devices"
46+
mode: embedded
47+
enable_offload: false
48+
frontend:
49+
type: dynamo
50+
enable_multiple_frontends: false
51+
args:
52+
router-mode: kv
53+
router-temperature: 0
54+
router-queue-threshold: 65536
55+
active-decode-blocks-threshold: None
56+
active-prefill-tokens-threshold: None
57+
active-prefill-tokens-threshold-frac: None
58+
tokenizer: fastokens
59+
engine:
60+
type: vllm
61+
connector: null
62+
roles:
63+
agg:
64+
nodes: 2
65+
workers: 1
66+
gpus: 8
67+
env:
68+
HF_HUB_CACHE: /hf_hub_cache
69+
HUGGINGFACE_HUB_CACHE: /hf_hub_cache
70+
TRANSFORMERS_CACHE: /hf_hub_cache
71+
VLLM_ENGINE_READY_TIMEOUT_S: '3600'
72+
VLLM_RPC_TIMEOUT: '600000'
73+
VLLM_LOG_STATS_INTERVAL: '1'
74+
VLLM_V2_WARMUP_MAX_NUM_SEQS: '16'
75+
TILELANG_CLEANUP_TEMP_FILES: '1'
76+
VLLM_USE_NCCL_SYMM_MEM: '0'
77+
TORCH_SYMMMEM: NVSHMEM
78+
NCCL_CUMEM_ENABLE: '1'
79+
NCCL_MNNVL_ENABLE: '1'
80+
NCCL_NVLS_ENABLE: '1'
81+
VLLM_SERVER_DEV_MODE: '1'
82+
VLLM_USE_V2_MODEL_RUNNER: '1'
83+
VLLM_USE_RUST_FRONTEND: '1'
84+
VLLM_ALLREDUCE_USE_FLASHINFER: '1'
85+
VLLM_FLASHINFER_ALLREDUCE_BACKEND: auto
86+
VLLM_MOONCAKE_LOAD_RECV_THREADS: '20'
87+
VLLM_PREFIX_CACHE_RETENTION_INTERVAL: '32768'
88+
VLLM_SPARSE_INDEXER_MAX_LOGITS_MB: '1024'
89+
VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS: '1800'
90+
UCX_MEMTYPE_CACHE: n
91+
UCX_NET_DEVICES: "@fabric.rdma-device-ports"
92+
UCX_TLS: rc,cuda_copy
93+
NCCL_IB_HCA: "@fabric.rdma-devices"
94+
NCCL_P2P_LEVEL: NVL
95+
MC_ENABLE_DEST_DEVICE_AFFINITY: '1'
96+
MC_STORE_CLIENT_METRIC: '1'
97+
MC_STORE_CLIENT_METRIC_INTERVAL: '5'
98+
MC_TE_METRIC: '0'
99+
DG_JIT_CACHE_DIR: /tmp/dg-cache-dsv4-gb200-dspark-agg-{job_id}
100+
VLLM_USE_BREAKABLE_CUDAGRAPH: '0'
101+
args:
102+
served-model-name: deepseek-ai/DeepSeek-V4-Pro-0813
103+
kv-cache-dtype: fp8
104+
tensor-parallel-size: 8
105+
pipeline-parallel-size: 1
106+
disable-custom-all-reduce: true
107+
attention-config: '{"backend":"FLASHINFER_MLA_SPARSE_DSV4","use_prefill_query_quantization":true,"indexer_kv_dtype":"mxfp4"}'
108+
max-model-len: 1048576
109+
max-num-seqs: 16
110+
max-num-batched-tokens: 8192
111+
trust-remote-code: true
112+
no-enable-flashinfer-autotune: true
113+
block-size: 256
114+
cudagraph-metrics: true
115+
compilation-config: '{"cudagraph_mode":"FULL_DECODE_ONLY","cudagraph_capture_sizes":[7,14,21,28,35,42,49,56,63,70,77,84,91,98,105,112],"mode":0}'
116+
speculative-config: '{"method":"dspark","num_speculative_tokens":6,"draft_sample_method":"probabilistic","rejection_sample_method":"block"}'
117+
gpu-memory-utilization: 0.94
118+
stream-interval: 10
119+
no-disable-hybrid-kv-cache-manager: true
120+
tokenizer-mode: deepseek_v4
121+
sbatch_directives:
122+
cpus-per-task: '144'
123+
mem: '0'
124+
srun_options:
125+
container-remap-root: ''
126+
cpu-bind: none
127+
benchmark:
128+
type: custom
129+
command: bash /infmax-workspace/benchmarks/srt_agentic.sh
130+
env:
131+
INFMAX_CONTAINER_WORKSPACE: /infmax-workspace
132+
PORT: '8000'
133+
AIPERF_USE_DYNAMO_CONV_AWARE_ROUTING: '0'
134+
WEKA_LOADER_OVERRIDE: semianalysis_cc_traces_weka_062126
135+
AIPERF_DATASET_MMAP_CACHE_ENABLED: 'false'

0 commit comments

Comments
 (0)