kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-04 18:21:52 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[https://nvbugs/5821433][fix] fix test_auto_scaling for 2 GPUs (#10866)

Browse Source 
Signed-off-by: Lizhi Zhou <1432185+reasonsolo@users.noreply.github.com>
Signed-off-by: Wangshanshan <30051912+dominicshanshan@users.noreply.github.com>
This commit is contained in:
Lizhi Zhou 2026-01-22 13:08:14 +08:00 committed by Yanchao Lu
parent 6c2ecad2fe
commit 4d282bd7c1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/integration/defs/disaggregated/test_auto_scaling.py
View File

@ -465,7 +465,7 @@ async def test_worker_restart(model_name, disagg_server_config, worker_config,
worker_config,
work_dir,
port=0,
device=2)
device=0)
await wait_for_disagg_server_status(disagg_port, 1, 1)
await asyncio.sleep(CHECK_STATUS_INTERVAL)
verify_cluster_info(True, 1, 1, port=disagg_port)
@ -487,7 +487,7 @@ async def test_worker_restart(model_name, disagg_server_config, worker_config,
worker_config,
work_dir,
port=0,
device=3)
device=1)
await wait_for_disagg_server_status(disagg_port, 1, 1)
await asyncio.sleep(CHECK_STATUS_INTERVAL)
verify_cluster_info(True, 1, 1, port=disagg_port)