Cache optimizations (#418)

This commit is contained in:
Liangsheng Yin
2024-05-13 12:47:13 +08:00
committed by GitHub
parent 562b8857d8
commit 39191c8515
8 changed files with 117 additions and 96 deletions
+4 -4
View File
@@ -5,7 +5,7 @@ import uvloop
import zmq
import zmq.asyncio
from sglang.srt.backend_config import GLOBAL_BACKEND_CONFIG
from sglang import global_config
from sglang.srt.managers.router.model_rpc import ModelRpcClient
from sglang.srt.server_args import PortArgs, ServerArgs
from sglang.srt.utils import get_exception_traceback
@@ -30,7 +30,7 @@ class RouterManager:
self.recv_reqs = []
# Init some configs
self.extend_dependency_time = GLOBAL_BACKEND_CONFIG.extend_dependency_time
self.request_dependency_time = global_config.request_dependency_time
async def loop_for_forward(self):
while True:
@@ -46,9 +46,9 @@ class RouterManager:
if len(out_pyobjs) != 0:
has_finished = any([obj.finished for obj in out_pyobjs])
if has_finished:
if self.extend_dependency_time > 0:
if self.request_dependency_time > 0:
slept = True
await asyncio.sleep(self.extend_dependency_time)
await asyncio.sleep(self.request_dependency_time)
if not slept:
await asyncio.sleep(0.0006)