Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
norm
vllm
Commits
2410e320
Unverified
Commit
2410e320
authored
Feb 27, 2024
by
Jingru
Committed by
GitHub
Feb 26, 2024
Browse files
fix `get_ip` error in pure ipv6 environment (#2931)
parent
48a8f4a7
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
19 additions
and
5 deletions
+19
-5
vllm/utils.py
vllm/utils.py
+19
-5
No files found.
vllm/utils.py
View file @
2410e320
...
...
@@ -162,9 +162,16 @@ def make_async(func: Callable[..., T]) -> Callable[..., Awaitable[T]]:
def
get_ip
()
->
str
:
# try ipv4
s
=
socket
.
socket
(
socket
.
AF_INET
,
socket
.
SOCK_DGRAM
)
s
.
connect
((
"8.8.8.8"
,
80
))
# Doesn't need to be reachable
return
s
.
getsockname
()[
0
]
try
:
s
.
connect
((
"dns.google"
,
80
))
# Doesn't need to be reachable
return
s
.
getsockname
()[
0
]
except
OSError
:
# try ipv6
s
=
socket
.
socket
(
socket
.
AF_INET6
,
socket
.
SOCK_DGRAM
)
s
.
connect
((
"dns.google"
,
80
))
return
s
.
getsockname
()[
0
]
def
get_distributed_init_method
(
ip
:
str
,
port
:
int
)
->
str
:
...
...
@@ -172,9 +179,16 @@ def get_distributed_init_method(ip: str, port: int) -> str:
def
get_open_port
()
->
int
:
with
socket
.
socket
(
socket
.
AF_INET
,
socket
.
SOCK_STREAM
)
as
s
:
s
.
bind
((
""
,
0
))
return
s
.
getsockname
()[
1
]
# try ipv4
try
:
with
socket
.
socket
(
socket
.
AF_INET
,
socket
.
SOCK_STREAM
)
as
s
:
s
.
bind
((
""
,
0
))
return
s
.
getsockname
()[
1
]
except
OSError
:
# try ipv6
with
socket
.
socket
(
socket
.
AF_INET6
,
socket
.
SOCK_STREAM
)
as
s
:
s
.
bind
((
""
,
0
))
return
s
.
getsockname
()[
1
]
def
set_cuda_visible_devices
(
device_ids
:
List
[
int
])
->
None
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment