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
67fc4268
Unverified
Commit
67fc4268
authored
Feb 28, 2025
by
Jee Jee Li
Committed by
GitHub
Feb 27, 2025
Browse files
[Misc] Print FusedMoE detail info (#13974)
parent
9804145c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
20 additions
and
0 deletions
+20
-0
vllm/model_executor/layers/fused_moe/layer.py
vllm/model_executor/layers/fused_moe/layer.py
+20
-0
No files found.
vllm/model_executor/layers/fused_moe/layer.py
View file @
67fc4268
...
...
@@ -737,3 +737,23 @@ class FusedMoE(torch.nn.Module):
# If we are in the row parallel case (down_proj)
else
:
param_data
[
expert_id
]
=
loaded_weight
def
extra_repr
(
self
)
->
str
:
s
=
(
f
"global_num_experts=
{
self
.
global_num_experts
}
, "
f
"local_num_experts=
{
self
.
local_num_experts
}
, "
f
"top_k=
{
self
.
top_k
}
, "
f
"intermediate_size_per_partition=
{
self
.
intermediate_size_per_partition
}
, "
# noqa: E501
f
"tp_size=
{
self
.
tp_size
}
,
\n
"
f
"ep_size=
{
self
.
ep_size
}
, "
f
"reduce_results=
{
self
.
reduce_results
}
, "
f
"renormalize=
{
self
.
renormalize
}
, "
f
"use_grouped_topk=
{
self
.
use_grouped_topk
}
"
)
if
self
.
use_grouped_topk
:
s
+=
f
", num_expert_group=
{
self
.
num_expert_group
}
, topk_group=
{
self
.
topk_group
}
"
# noqa: E501
s
+=
f
", scoring_func='
{
self
.
scoring_func
}
', activation='
{
self
.
activation
}
'"
# noqa: E501
return
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