Update eplb_device_transfer_loader.py

Mercykid-bash · web-flow · commit f75036d13793 · 2025-12-08T11:56:28.000+08:00
diff --git a/vllm_ascend/eplb/core/eplb_device_transfer_loader.py b/vllm_ascend/eplb/core/eplb_device_transfer_loader.py
@@ -50,6 +50,10 @@ def generate_expert_d2d_transfer_task(self, expert_send_info,
             )
             return
 
+        # If neither send nor receive task is needed for this layer on this rank, return
+        if not (expert_send_info or expert_recv_info):
+            return
+
         self.updated_expert_map = updated_expert_map
 
         self.layer_id = layer_id

Original file line number	Diff line number	Diff line change
`@@ -50,6 +50,10 @@ def generate_expert_d2d_transfer_task(self, expert_send_info,`
`50`	`50`	`)`
`51`	`51`	`return`
`52`	`52`
	`53`	`+ # If neither send nor receive task is needed for this layer on this rank, return`
	`54`	`+ if not (expert_send_info or expert_recv_info):`
	`55`	`+ return`
	`56`	`+`
`53`	`57`	`self.updated_expert_map = updated_expert_map`
`54`	`58`
`55`	`59`	`self.layer_id = layer_id`