Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
bdb903bb
Unverified
Commit
bdb903bb
authored
Mar 17, 2026
by
Wentao Ye
Committed by
GitHub
Mar 17, 2026
Browse files
[Bug] Fix FlashInfer MNNVL socket collisions under concurrent vLLM jobs (#36674)
Signed-off-by:
yewentao256
<
zhyanwentao@126.com
>
parent
68f783a7
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
16 additions
and
9 deletions
+16
-9
vllm/distributed/device_communicators/flashinfer_all_reduce.py
...distributed/device_communicators/flashinfer_all_reduce.py
+16
-9
No files found.
vllm/distributed/device_communicators/flashinfer_all_reduce.py
View file @
bdb903bb
...
...
@@ -3,6 +3,8 @@
import
atexit
import
os
import
random
import
threading
import
torch
...
...
@@ -67,15 +69,20 @@ def initialize_fi_ar_workspace(
backend
=
envs
.
VLLM_FLASHINFER_ALLREDUCE_BACKEND
comm_backend
=
TorchDistBackend
(
group
=
group
)
_fi_ar_workspace
=
flashinfer_comm
.
create_allreduce_fusion_workspace
(
backend
=
backend
,
world_size
=
world_size
,
rank
=
rank
,
max_token_num
=
max_token_num
,
hidden_dim
=
hidden_dim
,
dtype
=
dtype
,
comm_backend
=
comm_backend
,
)
rng_state
=
random
.
getstate
()
try
:
random
.
seed
(
int
.
from_bytes
(
os
.
urandom
(
16
),
byteorder
=
"big"
))
_fi_ar_workspace
=
flashinfer_comm
.
create_allreduce_fusion_workspace
(
backend
=
backend
,
world_size
=
world_size
,
rank
=
rank
,
max_token_num
=
max_token_num
,
hidden_dim
=
hidden_dim
,
dtype
=
dtype
,
comm_backend
=
comm_backend
,
)
finally
:
random
.
setstate
(
rng_state
)
assert
_fi_ar_workspace
is
not
None
logger
.
debug
(
"Initialized FlashInfer All Reduce workspace: backend=%s, "
...
...
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