[CI] Fix and wire encoder/manager cudagraph unit tests (#49339)
Signed-off-by: Nick Hill <nickhill123@gmail.com> Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Nick Hill committed
Jul 21, 2026 at 20:41 UTC
1dca300653e63db8d6d00b2073d97bd6d59d3fd7
3 files changed
+9
.buildkite/test_areas/cuda.yaml
+3
@@ -26,7 +26,10 @@ steps:
26
- vllm/v1/cudagraph_dispatcher.py
27
- vllm/config/compilation.py
28
- vllm/compilation
29
+ - vllm/v1/worker/encoder_cudagraph.py
30
+ - vllm/v1/worker/encoder_cudagraph_defs.py
31
commands:
32
- pytest -v -s v1/cudagraph/test_cudagraph_dispatch.py
33
- pytest -v -s v1/cudagraph/test_cudagraph_mode.py
34
- pytest -v -s v1/cudagraph/test_breakable_cudagraph.py
35
+ - pytest -v -s v1/cudagraph/test_encoder_cudagraph.py
.buildkite/test_areas/misc.yaml
+1
@@ -143,6 +143,7 @@ steps:
143
- pytest -v -s -m 'cpu_test' v1/core
144
- pytest -v -s v1/structured_output
145
- pytest -v -s v1/test_serial_utils.py
146
+ - pytest -v -s v1/cudagraph/test_cudagraph_manager.py
147
- pytest -v -s -m 'cpu_test' v1/kv_connector/unit
148
- pytest -v -s -m 'cpu_test' v1/metrics
149
tests/v1/cudagraph/test_encoder_cudagraph.py
+5
@@ -108,6 +108,11 @@ def _make_manager_with_budgets(budgets: list[int]) -> EncoderCudaGraphManager:
108
mgr.token_budgets = sorted(budgets)
109
mgr.max_batch_size = 16
110
mgr.use_dp = False
111
+ mgr.config = EncoderCudaGraphConfig(
112
+ modalities=["image"],
113
+ buffer_keys=[],
114
+ out_hidden_size=32,
115
+ )
116
mgr.budget_graphs = {"default": {}}
117
mgr.graph_pool = None
118
mgr.graph_hits = 0