MCPcopy
hub / github.com/ModelTC/LightLLM / _up_tokens_to_pd_master

Function _up_tokens_to_pd_master

lightllm/server/httpserver/pd_loop.py:180–186  ·  view source on GitHub ↗
(forwarding_queue: AsyncQueue, websocket)

Source from the content-addressed store, hash-verified

178
179# 转发token的task
180async def _up_tokens_to_pd_master(forwarding_queue: AsyncQueue, websocket):
181 while True:
182 handle_list = await forwarding_queue.wait_to_get_all_data()
183
184 if handle_list:
185 load_info: dict = _get_load_info()
186 await websocket.send(pickle.dumps((ObjType.TOKEN_PACKS, handle_list, load_info)))
187
188
189# 获取节点负载信息

Callers 1

_pd_handle_taskFunction · 0.85

Calls 3

_get_load_infoFunction · 0.85
wait_to_get_all_dataMethod · 0.80
sendMethod · 0.80

Tested by

no test coverage detected