Fix ZMQ bind error on non-zero rank nodes when using SGLANG_BLOCK_NONZERO_RANK_CHILDREN=0 (#13686)
This commit is contained in:
@@ -136,9 +136,12 @@ class Engine(EngineBase):
|
|||||||
|
|
||||||
# Initialize ZMQ sockets
|
# Initialize ZMQ sockets
|
||||||
context = zmq.Context(2)
|
context = zmq.Context(2)
|
||||||
|
if self.server_args.node_rank == 0:
|
||||||
self.send_to_rpc = get_zmq_socket(
|
self.send_to_rpc = get_zmq_socket(
|
||||||
context, zmq.DEALER, self.port_args.rpc_ipc_name, True
|
context, zmq.DEALER, self.port_args.rpc_ipc_name, True
|
||||||
)
|
)
|
||||||
|
else:
|
||||||
|
self.send_to_rpc = None
|
||||||
|
|
||||||
# Enable tracing
|
# Enable tracing
|
||||||
if server_args.enable_trace:
|
if server_args.enable_trace:
|
||||||
|
|||||||
Reference in New Issue
Block a user