Skip to content

Commit 975194f

Browse files
Klaud-Coldgithub-actions[bot]Oseltamivirclaude-fix-botclaude
authored
Update kimik2.5-fp4-b200-vllm vLLM image to v0.21.0 (#1395)
* $Update kimik2.5-fp4-b200-vllm vLLM image to v0.20.2\n\nRef #1154\n\nCo-authored-by: Klaud Cold <Klaud-Cold@users.noreply.github.com> * fix(kimik2.5_fp4_b200.sh): raise --gpu-memory-utilization 0.90 -> 0.98 vLLM v0.20.2's CUDA-graph memory profiling subtracts an aggressive chunk from the requested utilization, leaving negative space for the KV cache (-39.49 GiB observed). Raising to 0.98 gives the profiler enough headroom to land KV cache positive while still keeping ~2% as hard buffer. Alternative would have been setting VLLM_MEMORY_PROFILER_ESTIMATE_CUDAGRAPHS=0, but raising the cap is the minimum-blast-radius fix and matches what similar B200 recipes use. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com> * fix(kimik2.5_fp4_b200.sh): disable CUDA-graph memory estimator + restore 0.90 mem-util Raising --gpu-memory-utilization to 0.98 wasn't enough — vLLM v0.20.2's CUDA-graph memory profiler still pre-reserves ~57 GB/GPU upfront, leaving the effective utilization at ~0.66 and the KV cache at -25 GiB (engine won't start). Disable the estimator with VLLM_MEMORY_PROFILER_ESTIMATE_CUDAGRAPHS=0 (same pattern as benchmarks/single_node/agentic/kimik2.5_fp4_b200.sh:65) and revert --gpu-memory-utilization back to 0.90. The 0.90 reservation already leaves ~18 GB/GPU as the same safety net the estimator was trying to enforce. --------- Co-authored-by: claude[bot] <41898282+claude[bot]@users.noreply.github.com> Co-authored-by: Bryan Shan <58582368+Oseltamivir@users.noreply.github.com> Co-authored-by: claude-fix-bot <claude-fix-bot@local> Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com> Co-authored-by: functionstackx <47992694+functionstackx@users.noreply.github.com>
1 parent 82914ea commit 975194f

3 files changed

Lines changed: 15 additions & 1 deletion

File tree

.github/configs/nvidia-master.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2632,7 +2632,7 @@ kimik2.5-int4-h200-vllm-agentic:
26322632
- { tp: 8, offloading: cpu, conc-list: [6, 7, 8, 9, 10, 11, 12, 13, 14] }
26332633

26342634
kimik2.5-fp4-b200-vllm:
2635-
image: vllm/vllm-openai:v0.17.0
2635+
image: vllm/vllm-openai:v0.20.2
26362636
model: nvidia/Kimi-K2.5-NVFP4
26372637
model-prefix: kimik2.5
26382638
runner: b200

benchmarks/single_node/kimik2.5_fp4_b200.sh

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,13 @@ fi
3333
# Start GPU monitoring (power, temperature, clocks every second)
3434
start_gpu_monitor
3535

36+
# vLLM v0.20.2+'s CUDA-graph memory profiler pre-reserves ~57 GB/GPU upfront
37+
# (~32% of total), which collides with --gpu-memory-utilization=0.90 and
38+
# leaves negative space for the KV cache. Disable the profiler — our 0.90
39+
# already leaves ~18 GB/GPU as safety net (same pattern as
40+
# benchmarks/single_node/agentic/kimik2.5_fp4_b200.sh).
41+
export VLLM_MEMORY_PROFILER_ESTIMATE_CUDAGRAPHS=0
42+
3643
set -x
3744
vllm serve $MODEL --host 0.0.0.0 --port $PORT \
3845
--tensor-parallel-size=$TP \

perf-changelog.yaml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2704,3 +2704,10 @@
27042704
description:
27052705
- "Update vLLM image from v0.19.0-cu130 (25d old) to v0.21.0"
27062706
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/1448
2707+
2708+
- config-keys:
2709+
- kimik2.5-fp4-b200-vllm
2710+
description:
2711+
- "Update vLLM image from v0.20.2 to v0.21.0"
2712+
- "Add VLLM_MEMORY_PROFILER_ESTIMATE_CUDAGRAPHS=0 to disable aggressive CUDA-graph memory profiler that OOMs the KV cache"
2713+
pr-link: https://github.com/SemiAnalysisAI/InferenceX/pull/1395

0 commit comments

Comments
 (0)