Commit 4e10de4
fix(test): lower gpu_memory_utilization for inference validation step (vllm-project#727)
## Summary
Fixes the DFlash OOM in nightly CI
([INFERENG-8705](https://redhat.atlassian.net/browse/INFERENG-8705)).
### Root cause
PR vllm-project#495 refactored `vllm_kwargs` handling in `test_offline_training.py`
to spread a single kwargs dict into both `launch_vllm_server_context`
(data-gen server) and `run_vllm_engine` (inference validation). This
inadvertently raised the inference step's `gpu_memory_utilization` from
**0.8** (the `run_vllm_engine` default) to **0.9**, leaving insufficient
headroom for vLLM's warmup allocation during speculative decoding.
The warmup allocation is **4.64 GiB** (`max_num_seqs=1024 ×
(num_spec_tokens+1)=8 × vocab_size=151,936 × 4 bytes`), introduced by
vLLM PR [#37812](vllm-project/vllm#37812) which
added spec-decode-aware warmup. With `gpu_memory_utilization=0.9` on an
80 GiB H100, only ~4.5 GiB remains free after model + KV cache — not
enough for the 4.64 GiB warmup logits tensor.
The failure is intermittent because it also depends on a ~1.83 GiB
background GPU process present on some CI runners but not others.
### Fix
Override `gpu_memory_utilization` back to **0.8** for the inference
validation step only. The data-gen server keeps 0.9 since it doesn't
perform spec decode warmup.
## Test plan
- [ ] CI nightly matrix passes (DFlash test specifically)
- [ ] Verify locally: `pytest tests/e2e/smoke/test_offline_training.py
-k dflash`
🤖 Generated with [Claude Code](https://claude.com/claude-code)
Signed-off-by: Orestis Zambounis <orestis.zambounis@gmail.com>
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>1 parent 87d47ff commit 4e10de4
1 file changed
Lines changed: 2 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
165 | 165 | | |
166 | 166 | | |
167 | 167 | | |
| 168 | + | |
168 | 169 | | |
169 | 170 | | |
170 | 171 | | |
| |||
173 | 174 | | |
174 | 175 | | |
175 | 176 | | |
176 | | - | |
| 177 | + | |
177 | 178 | | |
0 commit comments