File tree Expand file tree Collapse file tree 1 file changed +0
-9
lines changed
vllm/model_executor/layers/fused_moe Expand file tree Collapse file tree 1 file changed +0
-9
lines changed Original file line number Diff line number Diff line change 1
1
# SPDX-License-Identifier: Apache-2.0
2
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
- import socket
4
- from contextlib import closing
5
3
from math import prod
6
4
from typing import Optional
7
5
@@ -101,13 +99,6 @@ def _fp8_perm(m: torch.Tensor, idx: torch.Tensor) -> torch.Tensor:
101
99
return m [idx , ...]
102
100
103
101
104
- def find_free_port ():
105
- with closing (socket .socket (socket .AF_INET , socket .SOCK_STREAM )) as s :
106
- s .bind (('' , 0 ))
107
- s .setsockopt (socket .SOL_SOCKET , socket .SO_REUSEADDR , 1 )
108
- return s .getsockname ()[1 ]
109
-
110
-
111
102
# TODO(bnell): better name
112
103
def maybe_fix_scales (scales : Optional [torch .Tensor ],
113
104
num_experts : int ) -> Optional [torch .Tensor ]:
You can’t perform that action at this time.
0 commit comments