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
ddb01cd9
Commit
ddb01cd9
authored
Dec 05, 2025
by
yangql
Browse files
add ALLOW_MNNV default falase, use VLLM_ALLOW_MNNVL=1
parent
52f895ab
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
10 additions
and
2 deletions
+10
-2
vllm/distributed/device_communicators/all2all.py
vllm/distributed/device_communicators/all2all.py
+4
-2
vllm/envs.py
vllm/envs.py
+6
-0
No files found.
vllm/distributed/device_communicators/all2all.py
View file @
ddb01cd9
...
...
@@ -8,7 +8,7 @@ import torch.distributed as dist
from
vllm.forward_context
import
get_forward_context
from
vllm.logger
import
init_logger
from
vllm.utils
import
has_deep_ep
,
has_pplx
import
vllm.envs
as
envs
from
.base_device_communicator
import
All2AllManagerBase
,
Cache
logger
=
init_logger
(
__name__
)
...
...
@@ -253,7 +253,9 @@ class DeepEPLLAll2AllManager(DeepEPAll2AllManagerBase):
num_nvl_bytes
=
num_nvl_bytes
,
num_rdma_bytes
=
num_rdma_bytes
,
low_latency_mode
=
True
,
num_qps_per_rank
=
num_qps_per_rank
)
num_qps_per_rank
=
num_qps_per_rank
,
allow_mnnvl
=
envs
.
VLLM_ALLOW_MNNVL
,
)
def
get_handle
(
self
,
kwargs
):
"""
...
...
vllm/envs.py
View file @
ddb01cd9
...
...
@@ -128,6 +128,7 @@ if TYPE_CHECKING:
VLLM_NIXL_SIDE_CHANNEL_HOST
:
str
=
"localhost"
VLLM_NIXL_SIDE_CHANNEL_PORT
:
int
=
5557
VLLM_ALL2ALL_BACKEND
:
str
=
"naive"
VLLM_ALLOW_MNNVL
:
bool
=
False
VLLM_MAX_TOKENS_PER_EXPERT_FP4_MOE
:
int
=
163840
VLLM_TOOL_PARSE_REGEX_TIMEOUT_SECONDS
:
int
=
1
VLLM_SLEEP_WHEN_IDLE
:
bool
=
False
...
...
@@ -951,6 +952,11 @@ environment_variables: dict[str, Callable[[], Any]] = {
# - "mori", use mori kernels
"VLLM_ALL2ALL_BACKEND"
:
lambda
:
os
.
getenv
(
"VLLM_ALL2ALL_BACKEND"
,
"naive"
),
# use ALLOW_MNNVL
"VLLM_ALLOW_MNNVL"
:
lambda
:
(
os
.
environ
.
get
(
"VLLM_ALLOW_MNNVL"
,
"False"
).
lower
()
in
(
"true"
,
"1"
)),
# Control the maximum number of tokens per expert supported by the
# NVFP4 MoE CUTLASS Kernel. This value is used to create a buffer for
...
...
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