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
66b809cc
Commit
66b809cc
authored
Feb 08, 2025
by
zhuwenwen
Browse files
Merge tag 'v0.7.2' into v0.7.2-dev
parents
37b63c24
0408efc6
Changes
1000
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
36 additions
and
0 deletions
+36
-0
tests/core/test_num_computed_tokens_update.py
tests/core/test_num_computed_tokens_update.py
+2
-0
tests/core/test_scheduler.py
tests/core/test_scheduler.py
+2
-0
tests/core/test_scheduler_encoder_decoder.py
tests/core/test_scheduler_encoder_decoder.py
+2
-0
tests/core/test_serialization.py
tests/core/test_serialization.py
+2
-0
tests/core/utils.py
tests/core/utils.py
+2
-0
tests/distributed/test_ca_buffer_sharing.py
tests/distributed/test_ca_buffer_sharing.py
+2
-0
tests/distributed/test_comm_ops.py
tests/distributed/test_comm_ops.py
+1
-0
tests/distributed/test_custom_all_reduce.py
tests/distributed/test_custom_all_reduce.py
+2
-0
tests/distributed/test_distributed_oot.py
tests/distributed/test_distributed_oot.py
+2
-0
tests/distributed/test_multi_node_assignment.py
tests/distributed/test_multi_node_assignment.py
+1
-0
tests/distributed/test_pipeline_parallel.py
tests/distributed/test_pipeline_parallel.py
+1
-0
tests/distributed/test_pipeline_partition.py
tests/distributed/test_pipeline_partition.py
+2
-0
tests/distributed/test_pp_cudagraph.py
tests/distributed/test_pp_cudagraph.py
+2
-0
tests/distributed/test_pynccl.py
tests/distributed/test_pynccl.py
+2
-0
tests/distributed/test_same_node.py
tests/distributed/test_same_node.py
+2
-0
tests/distributed/test_shm_broadcast.py
tests/distributed/test_shm_broadcast.py
+2
-0
tests/distributed/test_torchrun_example.py
tests/distributed/test_torchrun_example.py
+2
-0
tests/distributed/test_utils.py
tests/distributed/test_utils.py
+2
-0
tests/encoder_decoder/test_e2e_correctness.py
tests/encoder_decoder/test_e2e_correctness.py
+1
-0
tests/engine/output_processor/test_multi_step.py
tests/engine/output_processor/test_multi_step.py
+2
-0
No files found.
Too many changes to show.
To preserve performance only
1000 of 1000+
files are displayed.
Plain diff
Email patch
tests/core/test_num_computed_tokens_update.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
os
import
pytest
...
...
tests/core/test_scheduler.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
time
from
collections
import
deque
from
typing
import
List
,
Set
,
Tuple
...
...
tests/core/test_scheduler_encoder_decoder.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
from
typing
import
List
import
pytest
# noqa
...
...
tests/core/test_serialization.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
msgspec
from
vllm.executor.msgspec_utils
import
decode_hook
,
encode_hook
...
...
tests/core/utils.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
time
from
collections
import
defaultdict
from
typing
import
Any
,
Dict
,
List
,
Optional
...
...
tests/distributed/test_ca_buffer_sharing.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
# can only run on machines with p2p access across GPUs
# can only run with torchrun:
# torchrun --nproc_per_node=2 tests/distributed/test_ca_buffer_sharing.py
...
...
tests/distributed/test_comm_ops.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
"""Test the communication operators.
Run `pytest tests/distributed/test_comm_ops.py`.
...
...
tests/distributed/test_custom_all_reduce.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
os
import
random
...
...
tests/distributed/test_distributed_oot.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
from
..entrypoints.openai.test_oot_registration
import
(
run_and_test_dummy_opt_api_server
)
...
...
tests/distributed/test_multi_node_assignment.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
"""Make sure ray assigns GPU workers to the correct node.
Run:
...
...
tests/distributed/test_pipeline_parallel.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
"""
WARNING: This test runs in both single-node (4 GPUs) and multi-node
(2 node with 2 GPUs each) modes. If the test only uses 2 GPUs, it is
...
...
tests/distributed/test_pipeline_partition.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
os
import
pytest
...
...
tests/distributed/test_pp_cudagraph.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
os
import
pytest
...
...
tests/distributed/test_pynccl.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
multiprocessing
import
os
from
typing
import
Dict
,
List
...
...
tests/distributed/test_same_node.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
os
import
torch.distributed
as
dist
...
...
tests/distributed/test_shm_broadcast.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
multiprocessing
import
random
import
time
...
...
tests/distributed/test_torchrun_example.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
# unit test for `examples/offline_inference/torchrun_example.py`
import
random
...
...
tests/distributed/test_utils.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
socket
import
pytest
...
...
tests/encoder_decoder/test_e2e_correctness.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
"""E2E tests to verify the correctness of the encoder-decoder framework
Run `pytest tests/encoder_decoder/test_e2e_correctness.py`.
...
...
tests/engine/output_processor/test_multi_step.py
View file @
66b809cc
# SPDX-License-Identifier: Apache-2.0
import
random
from
unittest.mock
import
MagicMock
...
...
Prev
1
…
5
6
7
8
9
10
11
12
13
…
50
Next
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