Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
8c6f694a
Unverified
Commit
8c6f694a
authored
Aug 22, 2024
by
youkaichao
Committed by
GitHub
Aug 22, 2024
Browse files
[ci] refine dependency for distributed tests (#7776)
parent
eeee1c3b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
17 additions
and
8 deletions
+17
-8
.buildkite/test-pipeline.yaml
.buildkite/test-pipeline.yaml
+17
-8
No files found.
.buildkite/test-pipeline.yaml
View file @
8c6f694a
...
...
@@ -95,7 +95,8 @@ steps:
num_gpus
:
4
fast_check
:
true
source_file_dependencies
:
-
vllm/
-
vllm/distributed/
-
vllm/core/
-
tests/distributed
-
tests/spec_decode/e2e/test_integration_dist_tp4
commands
:
...
...
@@ -284,8 +285,11 @@ steps:
num_gpus
:
2
num_nodes
:
2
source_file_dependencies
:
-
vllm/
-
tests/distributed/test_same_node
-
vllm/distributed/
-
vllm/engine/
-
vllm/executor/
-
vllm/model_executor/models/
-
tests/distributed/
commands
:
-
# the following commands are for the first node, with ip 192.168.10.10 (ray environment already set up)
-
VLLM_TEST_SAME_HOST=0 torchrun --nnodes 2 --nproc-per-node=2 --rdzv_backend=c10d --rdzv_endpoint=192.168.10.10 distributed/test_same_node.py
...
...
@@ -298,8 +302,11 @@ steps:
working_dir
:
"
/vllm-workspace/tests"
num_gpus
:
2
source_file_dependencies
:
-
vllm/
-
tests/distributed
-
vllm/distributed/
-
vllm/engine/
-
vllm/executor/
-
vllm/model_executor/models/
-
tests/distributed/
commands
:
-
VLLM_TEST_SAME_HOST=1 torchrun --nproc-per-node=4 distributed/test_same_node.py
-
TARGET_TEST_SUITE=L4 pytest -v -s distributed/test_basic_distributed_correctness.py
...
...
@@ -333,9 +340,11 @@ steps:
working_dir
:
"
/vllm-workspace/tests"
num_gpus
:
4
source_file_dependencies
:
-
vllm/
-
tests/distributed/test_pp_cudagraph.py
-
tests/distributed/test_pipeline_parallel
-
vllm/distributed/
-
vllm/engine/
-
vllm/executor/
-
vllm/model_executor/models/
-
tests/distributed/
commands
:
-
pytest -v -s distributed/test_pp_cudagraph.py
-
pytest -v -s distributed/test_pipeline_parallel.py
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment