[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:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user