chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

[diffusion] refactor: change zmq socket type to router for scheduler (#15479)

Browse Source
This commit is contained in:
Mick
2025-12-21 00:39:54 +08:00
committed by GitHub
parent bee8ac5b88
commit bc18cb8631
16 changed files with 310 additions and 305 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/sglang/multimodal_gen/runtime/utils/common.py
View File

@@ -170,13 +170,7 @@ def get_zmq_socket(
set_send_opt()
elif socket_type == zmq.PULL:
set_recv_opt()
elif socket_type == zmq.DEALER:
set_send_opt()
set_recv_opt()
elif socket_type == zmq.REQ:
set_send_opt()
set_recv_opt()
elif socket_type == zmq.REP:
elif socket_type in [zmq.DEALER, zmq.REQ, zmq.REP, zmq.ROUTER]:
set_send_opt()
set_recv_opt()
else:

2
python/sglang/multimodal_gen/runtime/utils/perf_logger.py
View File

@@ -168,7 +168,7 @@ class StageProfiler:
if self.simple_log:
self.logger.info(
f"[{self.stage_name}] finished in {execution_time_s:.4f} seconds"
f"[{self.stage_name}] finished in {execution_time_s:.4f} seconds",
)
if self.metrics_enabled and self.timings: