minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion]: fix scheduler crash on ZMQ messages with unexpected frame counts (#17890)

Browse Source 
Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
This commit is contained in:
Ratish P
2026-02-17 09:45:05 +08:00
committed by GitHub
co-authored by Xiaoyu Zhang
parent f9c3def7fe
commit 50ca24aebb
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/managers/scheduler.py
+7 -3
View File
@@ -241,9 +241,13 @@ class Scheduler:
if self.receiver is not None: if self.receiver is not None:
try: try:
try: try:
identity, _, payload = self.receiver.recv_multipart(zmq.NOBLOCK) # Accept valid REQ envelopes only, ignore malformed/probe frames.
recv_reqs = pickle.loads(payload) parts = self.receiver.recv_multipart(zmq.NOBLOCK)
except zmq.Again: identity, payload = parts[0], parts[-1]
# Ignore malformed probes or non-pickle data
recv_reqs = pickle.loads(payload) if len(parts) > 2 else []
except (zmq.Again, pickle.UnpicklingError, IndexError, EOFError):
recv_reqs = [] recv_reqs = []
except zmq.ZMQError: except zmq.ZMQError:
# re-raise or handle appropriately to let the outer loop continue # re-raise or handle appropriately to let the outer loop continue