Skip to content

GitLab

Commit d08e3d52 authored by 王敏's avatar 王敏
Browse files

去掉调试代码

parent 289f98c6
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 8 deletions
vllm/model_executor/layers/fused_moe/layer.py
View file @ d08e3d52
...@@ -281,9 +281,6 @@ def maybe_roundup_hidden_size( ...@@ -281,9 +281,6 @@ def maybe_roundup_hidden_size(
return hidden_size return hidden_size
first = True
# --8<-- [start:fused_moe] # --8<-- [start:fused_moe]
@CustomOp.register("fused_moe") @CustomOp.register("fused_moe")
class FusedMoE(CustomOp): class FusedMoE(CustomOp):
...@@ -401,11 +398,6 @@ class FusedMoE(CustomOp): ...@@ -401,11 +398,6 @@ class FusedMoE(CustomOp):
# Expert mapping used in self.load_weights # Expert mapping used in self.load_weights
self.expert_mapping = expert_mapping self.expert_mapping = expert_mapping
global first
if first:
print(f"###################self.global_num_experts:{self.global_num_experts}, self.logical_num_experts:{self.logical_num_experts} self.moe_parallel_config:{self.moe_parallel_config}")
first = False
# Round up hidden size if needed. # Round up hidden size if needed.
hidden_size = maybe_roundup_hidden_size( hidden_size = maybe_roundup_hidden_size(
hidden_size, hidden_size,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment