Skip to content

Commit 1547810

Browse files
author
lt
committed
delete empty tensor judgement
1 parent e7b7186 commit 1547810

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

vllm_ascend/eplb/eplb_updator.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,8 +237,6 @@ def unpack_update_batch(self, packed_update_info):
237237
def get_expert_load(self) -> torch.Tensor:
238238
expert_maps = self.shared_dict["expert_maps"]
239239
moe_load = self.shared_dict["moe_load"] # Tensor [L, W, global_experts_num]
240-
if not moe_load:
241-
return None
242240
num_local_experts = expert_maps.max() + 1
243241
load_info, _ = ExpertMapUtils.global2local_load(moe_load, expert_maps, num_local_experts)
244242
return load_info

0 commit comments

Comments
 (0)