Skip to content

Commit 3c17044

Browse files
noaux_eplb
1 parent 0b9f82b commit 3c17044

File tree

1 file changed

+0
-2
lines changed
  • fastdeploy/model_executor/layers/moe

1 file changed

+0
-2
lines changed

fastdeploy/model_executor/layers/moe/ep.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -430,8 +430,6 @@ def moe_select(self, layer: nn.Layer, gate_out: paddle.Tensor):
430430
expert_in_rank_num_list,
431431
tokens_per_expert_stats_list,
432432
) = layer.redundant_table_manger.get_ep_rank_to_expert_id_list_by_layer(layer.layer_idx)
433-
if layer.is_rearrange is False:
434-
expert_id_to_ep_rank_array = paddle.arange(layer.num_experts).cast("int32")
435433

436434
if layer.topk_method == "noaux_tc":
437435
from .moe import get_moe_scores

0 commit comments

Comments
 (0)