[CI] Enroll five steps in automatic sharding (#60492)
Signed-off-by: Thang Nguyen <thangnguyenvn647@gmail.com> Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com> Co-authored-by: khluu <khluu000@gmail.com>
This commit is contained in:
co-authored by
Claude Opus 5.5
khluu
parent
4cf1307e72
commit
7d47ac2ddb
@@ -231,6 +231,7 @@ steps:
|
||||
- label: ":nvidia: (H200 MIG 35GB) Entrypoints Integration (Pooling)"
|
||||
device: h200_35gb
|
||||
key: entrypoints-integration-pooling
|
||||
automatic_shard: true
|
||||
timeout_in_minutes: 75
|
||||
working_dir: "/vllm-workspace/tests"
|
||||
source_file_dependencies:
|
||||
|
||||
@@ -504,6 +504,7 @@ steps:
|
||||
- label: ":nvidia: (H200 MIG 35GB) Mamba Kernels"
|
||||
device: h200_35gb
|
||||
key: kernels-mamba-test
|
||||
automatic_shard: true
|
||||
timeout_in_minutes: 60
|
||||
source_file_dependencies:
|
||||
- csrc/mamba/
|
||||
|
||||
@@ -127,6 +127,7 @@ steps:
|
||||
- label: ":nvidia: (H200 MIG 35GB) Quantized Models"
|
||||
device: h200_35gb
|
||||
key: quantized-models-test
|
||||
automatic_shard: true
|
||||
timeout_in_minutes: 65
|
||||
env:
|
||||
VLLM_USE_V2_MODEL_RUNNER: "0"
|
||||
|
||||
@@ -5,6 +5,7 @@ steps:
|
||||
- label: ":nvidia: (H200 MIG 35GB) V1 Spec Decode"
|
||||
device: h200_35gb
|
||||
key: v1-spec-decode
|
||||
automatic_shard: true
|
||||
timeout_in_minutes: 40
|
||||
source_file_dependencies:
|
||||
- vllm/config/
|
||||
|
||||
Reference in New Issue
Block a user