| 1 | group: CUDA |
| 2 | depends_on: |
| 3 | - image-build |
| 4 | steps: |
| 5 | - label: ":nvidia: (H200 MIG 18GB) CUDA Platform" |
| 6 | key: platform-tests |
| 7 | timeout_in_minutes: 20 |
| 8 | device: h200_18gb |
| 9 | source_file_dependencies: |
| 10 | - vllm/envs.py |
| 11 | - vllm/logger.py |
| 12 | - vllm/platforms/ |
| 13 | - vllm/plugins/ |
| 14 | - vllm/utils/ |
| 15 | - tests/cuda |
| 16 | commands: |
| 17 | - pytest -v -s cuda/test_cuda_context.py |
| 18 | - pytest -v -s cuda/test_platform_no_cuda_init.py |
| 19 | - pytest -v -s cuda/test_cuda_compatibility_path.py |
| 20 | mirror: |
| 21 | amd: |
| 22 | label: ":amd: (MI355 DPX) CUDA Platform" |
| 23 | dind: false |
| 24 | device: mi355_dpx |
| 25 | working_dir: "/vllm-workspace/tests" |
| 26 | timeout_in_minutes: 25 |
| 27 | depends_on: |
| 28 | - image-build-amd |
| 29 | source_file_dependencies: |
| 30 | - vllm/ |
| 31 | - tests/cuda |
| 32 | |
| 33 | - label: ":nvidia: (H200 MIG 18GB) CUDAGraph" |
| 34 | device: h200_18gb |
| 35 | key: cudagraph |
| 36 | timeout_in_minutes: 30 |
| 37 | source_file_dependencies: |
| 38 | - tests/v1/cudagraph |
| 39 | - vllm/v1/cudagraph_dispatcher.py |
| 40 | - vllm/config/compilation.py |
| 41 | - vllm/compilation |
| 42 | - vllm/v1/worker/encoder_cudagraph.py |
| 43 | - vllm/v1/worker/encoder_cudagraph_defs.py |
| 44 | commands: |
| 45 | - pytest -v -s v1/cudagraph/test_cudagraph_dispatch.py |
| 46 | - pytest -v -s v1/cudagraph/test_cudagraph_mode.py |
| 47 | - pytest -v -s v1/cudagraph/test_breakable_cudagraph.py |
| 48 | - pytest -v -s v1/cudagraph/test_encoder_cudagraph.py |
| 49 | mirror: |
| 50 | amd: |
| 51 | label: ":amd: (MI250) CUDAGraph" |
| 52 | dind: false |
| 53 | device: mi250_1 |
| 54 | timeout_in_minutes: 55 |
| 55 | depends_on: |
| 56 | - image-build-amd |
| 57 | source_file_dependencies: |
| 58 | - tests/v1/cudagraph |
| 59 | - vllm/v1/cudagraph_dispatcher.py |
| 60 | - vllm/config/compilation.py |
| 61 | - vllm/compilation |
| 62 | - vllm/platforms/rocm.py |