Skip to content

Commit 8fc571e

Browse files
authored
[MRV2] Enable PP CUDA graph test (vllm-project#37830)
Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>
1 parent 3e2cde0 commit 8fc571e

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

.buildkite/test_areas/model_runner_v2.yaml

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,13 +87,12 @@ steps:
8787
- vllm/v1/worker/gpu/
8888
- vllm/v1/worker/gpu_worker.py
8989
- tests/distributed/test_pipeline_parallel.py
90-
#- tests/distributed/test_pp_cudagraph.py
90+
- tests/distributed/test_pp_cudagraph.py
9191
commands:
9292
- set -x
9393
- export VLLM_USE_V2_MODEL_RUNNER=1
9494
- pytest -v -s distributed/test_pipeline_parallel.py -k "not ray and not Jamba"
95-
# TODO: Uncomment once https://github.com/vllm-project/vllm/pull/35162 is merged.
96-
#- pytest -v -s distributed/test_pp_cudagraph.py -k "not ray"
95+
- pytest -v -s distributed/test_pp_cudagraph.py -k "not ray"
9796

9897
- label: Model Runner V2 Spec Decode
9998
timeout_in_minutes: 30

0 commit comments

Comments
 (0)