[Bugfix] Quick hot fix for nightly CI (#4727)

Quick fix for multi-node tests

---------

Signed-off-by: wangli <wangli858794774@gmail.com>
This commit is contained in:
Li Wang
2025-12-04 23:51:16 +08:00
committed by GitHub
parent ad0607f900
commit cd8e5be7c7
2 changed files with 30 additions and 29 deletions

View File

@@ -100,7 +100,7 @@ async def test_multi_node() -> None:
disaggregated_prefill = config.disaggregated_prefill
server_port = config.server_port
proxy_port = config.proxy_port
server_host = config.node_info.ip
server_host = config.master_ip
proxy_script = config.envs.get("DISAGGREGATED_PREFILL_PROXY_SCRIPT", \
'examples/disaggregated_prefill_v1/load_balance_proxy_server_example.py')
with config.launch_server_proxy(proxy_script):
@@ -123,8 +123,8 @@ async def test_multi_node() -> None:
run_aisbench_cases(local_model_path,
port,
aisbench_cases,
host_ip=config.cluster_ips[0])
host_ip=config.master_ip)
else:
# for the nodes except master, should hang until the task complete
master_url = f"http://{config.cluster_ips[0]}:{server_port}/health"
master_url = f"http://{config.master_ip}:{server_port}/health"
remote_server.hang_until_terminated(master_url)