Skip to content

GitLab

Unverified Commit 8bbb7c7f authored by Andreas Karatzas's avatar Andreas Karatzas Committed by GitHub
Browse files

[ROCm][CI][PD] Add Hybrid SSM integration tests to CI (#37924) 


Signed-off-by: default avatarAndreas Karatzas <akaratza@amd.com>
parent af945615
Show whitespace changes
Inline Side-by-side
Showing with 15 additions and 0 deletions
.buildkite/test-amd.yaml
View file @ 8bbb7c7f
...@@ -1387,6 +1387,21 @@ steps: ...@@ -1387,6 +1387,21 @@ steps:
- CROSS_LAYERS_BLOCKS=True ROCM_ATTN=1 bash v1/kv_connector/nixl_integration/config_sweep_accuracy_test.sh - CROSS_LAYERS_BLOCKS=True ROCM_ATTN=1 bash v1/kv_connector/nixl_integration/config_sweep_accuracy_test.sh
- label: Hyrbid SSM NixlConnector PD accuracy tests (4 GPUs) # TBD
timeout_in_minutes: 180
mirror_hardwares: [amdexperimental, amdproduction, amdgfx90anightly, amdmi250]
agent_pool: mi250_4
num_gpus: 4
working_dir: "/vllm-workspace/tests"
source_file_dependencies:
- vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py
- tests/v1/kv_connector/nixl_integration/
- vllm/platforms/rocm.py
commands:
- uv pip install --system -r /vllm-workspace/requirements/kv_connectors_rocm.txt
- HYBRID_SSM=1 ROCM_ATTN=1 bash v1/kv_connector/nixl_integration/config_sweep_accuracy_test.sh
- label: Distributed Tests (2 GPUs)(H100-MI250) # TBD - label: Distributed Tests (2 GPUs)(H100-MI250) # TBD
timeout_in_minutes: 180 timeout_in_minutes: 180
mirror_hardwares: [amdexperimental, amdproduction, amdgfx90anightly, amdmi250] mirror_hardwares: [amdexperimental, amdproduction, amdgfx90anightly, amdmi250]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment