Skip to content

Commit c6bdc59

Browse files
author
offline0806
committed
[EPLB]Fix ci.
Signed-off-by: offline0806 <z00858301@china.huawei.com>
1 parent 64cd7bd commit c6bdc59

File tree

2 files changed

+7
-8
lines changed

2 files changed

+7
-8
lines changed

vllm_ascend/eplb/core/eplb_utils.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -69,10 +69,9 @@ def generate_log2phy_map(expert_map):
6969
num_rank_holding_expert = positive_rank_idx.size(0)
7070

7171
if num_rank_holding_expert == 0:
72-
log2phy_map[:, idx] = torch.full(
73-
(num_ranks,),
74-
0,
75-
dtype=log2phy_map.dtype)
72+
log2phy_map[:, idx] = torch.full((num_ranks,),
73+
0,
74+
dtype=log2phy_map.dtype)
7675

7776
if num_rank_holding_expert == 1:
7877
log2phy_map[negative_rank_idx, idx] = torch.full(
@@ -84,8 +83,9 @@ def generate_log2phy_map(expert_map):
8483
random.choice(log2phy_map[positive_rank_idx, idx])
8584
for _ in range(num_ranks - num_rank_holding_expert)
8685
]
87-
log2phy_map[negative_rank_idx, idx] = torch.tensor(random_list,
88-
dtype=log2phy_map.dtype)
86+
log2phy_map[negative_rank_idx,
87+
idx] = torch.tensor(random_list,
88+
dtype=log2phy_map.dtype)
8989

9090
return log2phy_map
9191

vllm_ascend/ops/fused_moe.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -443,8 +443,7 @@ def forward(self,
443443
tuple) and len(e_hidden_states) == 2:
444444
e_hidden_states, shared_hidden_states = e_hidden_states
445445

446-
if isinstance(e_hidden_states,
447-
tuple) and len(e_hidden_states) == 3:
446+
if isinstance(e_hidden_states, tuple) and len(e_hidden_states) == 3:
448447
e_hidden_states, group_list_type, expert_tokens = e_hidden_states
449448

450449
if self.dynamic_eplb:

0 commit comments

Comments
 (0)