7f73c28a24
### What this PR does / why we need it?
This pull request mainly do the following things:
1. Add a doc for multi-node CI, The main content is the mechanism
principle and how to contribute
2. Simplify the config yaml for more developer-friendly
3. Optimized the mooncake installation script to prevent accidental
failures during installation
4. Fix the workflow to ensure the kubernetes can be apply correctly
5. Add Qwen3-235B-W8A8 disaggregated_prefill test
6. Add GLM-4.5 multi dp test
7. Add 2p1d 4nodes disaggregated_prefill test
8. Refactor nightly tests
### Does this PR introduce _any_ user-facing change?
### How was this patch tested?
- vLLM version: v0.11.0rc3
- vLLM main:
17c540a993
---------
Signed-off-by: wangli <wangli858794774@gmail.com>
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
from tests.e2e.conftest import RemoteOpenAIServer
|
|
from tests.e2e.nightly.multi_node.config.multi_node_config import (
|
|
DISAGGREGATED_PREFILL_PROXY_SCRIPT, MultiNodeConfig)
|
|
|
|
|
|
def test_multi_node() -> None:
|
|
config = MultiNodeConfig.from_yaml()
|
|
env_dict = config.envs
|
|
# perf_cmd = config.perf_cmd
|
|
# acc_cmd = config.acc_cmd
|
|
nodes_info = config.nodes_info
|
|
disaggregated_prefill = config.disaggregated_prefill
|
|
server_port = config.server_port
|
|
proxy_port = config.proxy_port
|
|
server_host = config.cluster_ips[0]
|
|
with config.launch_server_proxy(DISAGGREGATED_PREFILL_PROXY_SCRIPT):
|
|
with RemoteOpenAIServer(
|
|
model=config.model,
|
|
vllm_serve_args=config.server_cmd,
|
|
server_port=server_port,
|
|
server_host=server_host,
|
|
env_dict=env_dict,
|
|
auto_port=False,
|
|
proxy_port=proxy_port,
|
|
disaggregated_prefill=disaggregated_prefill,
|
|
nodes_info=nodes_info,
|
|
max_wait_seconds=2000,
|
|
) as remote_server:
|
|
# base_url = remote_server.url_root
|
|
if config.is_master:
|
|
pass
|
|
# TODO: enable perf and acc test
|
|
# subprocess.run(perf_cmd, check=True)
|
|
# subprocess.run(acc_cmd, check=True)
|
|
else:
|
|
remote_server.hang_until_terminated()
|