Skip to content

Commit cc35a75

Browse files
author
g30074156
committed
dynamic eplb
1 parent 8f9c3b7 commit cc35a75

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

vllm_ascend/eplb/core/policy/dynamic_ep_v2.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -658,9 +658,6 @@ def rebalance_experts(self, current_expert_table, expert_workload, is_node_redun
658658

659659
max_heat_per_layer_after[layer] = max(result, key=lambda x: x['total_load'])['total_load']
660660

661-
# for box in result:
662-
# print("before: ",
663-
# f"Box {box['device_id']}: Items = {box['assigned_experts']}, weight = {box['expert_weights']}, Total Weight = {box['total_load']}, Item Count = {box['expert_count']}")
664661
layer_changed_ratio = []
665662
for layer_idx in range(layer_num):
666663
layer_changed_ratio.append(self.safe_divide(max_heat_per_layer_after[layer_idx], max_heat_per_layer_before[layer_idx]))

0 commit comments

Comments
 (0)