mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-06-02 13:11:19 +08:00
updatedd
Signed-off-by: Robert Shaw <robshaw@redhat.com>
This commit is contained in:
parent
2a68433a82
commit
1ced153eec
@ -961,6 +961,8 @@ class DPAsyncMPClient(AsyncMPClient):
|
|||||||
counts, wave, running = msgspec.msgpack.decode(buf)
|
counts, wave, running = msgspec.msgpack.decode(buf)
|
||||||
self.current_wave = wave
|
self.current_wave = wave
|
||||||
self.engines_running = running
|
self.engines_running = running
|
||||||
|
# NOTE: counts includes num running for all global
|
||||||
|
# EngineCores, so need to slide for the local ones.
|
||||||
self.lb_engines = counts[dp_start_rank:dp_end_rank]
|
self.lb_engines = counts[dp_start_rank:dp_end_rank]
|
||||||
|
|
||||||
resources.stats_update_task = asyncio.create_task(
|
resources.stats_update_task = asyncio.create_task(
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user