From 3b6d539f6386258dc70663f159cf37c2832ce512 Mon Sep 17 00:00:00 2001
From: Lucius <souzou@foxmail.com>
Date: Mon, 21 Apr 2025 15:44:11 +0800
Subject: [PATCH] [Fix] Enhance DP Attention for IPv6 Compatibility (#4937)

---
 python/sglang/srt/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/sglang/srt/utils.py b/python/sglang/srt/utils.py
index 4151968dd..b93c28c39 100644
--- a/python/sglang/srt/utils.py
+++ b/python/sglang/srt/utils.py
@@ -942,6 +942,8 @@ def get_zmq_socket(
         buf_size = -1
 
     socket = context.socket(socket_type)
+    if endpoint.find("[") != -1:
+        socket.setsockopt(zmq.IPV6, 1)
 
     def set_send_opt():
         socket.setsockopt(zmq.SNDHWM, 0)